2017-10-11 66 views
2

我已經看到研究人員正在爲Caffe的原始版本添加一些功能,並根據他們需要的功能使用這些層和功能,然後通過Github共享這些版本。如果我沒有弄錯,有兩種方法:1)通過在添加C++和Cuda版本的圖層後重新編譯Caffe。 2)爲該功能編寫python代碼,並將其作爲Caffe中的python層調用。如何開始爲我的圖層編寫代碼?

我想根據我的研究問題向Caffe添加一個新圖層。我真的不應該從哪一點開始編寫新圖層以及我應該考慮哪些步驟。

我的問題是:

1)是否有任何文檔,或者說我可以用它編寫層中的任何學習資料?

2)上述哪種添加新圖層的方法是首選?

我非常感謝所有幫助和指導

非常感謝

回答

2

出於研究目的,「打轉轉」,它通常是更方便地編寫Python層:爲您節省編制等的喧囂
你可以在"Python"here找到一個簡短的教程。另一方面,如果你想要更好的性能,你應該爲你的圖層編寫一個本地C++代碼。
你可以找到關於它的簡短說明here

+1

非常感謝您的指導 –