2017-08-29 187 views
1

如果您使用的是自定義python圖層 - 並且假設您在python中正確編寫了該類 - 假設該類的名稱爲"my_ugly_custom_layer";並在linux命令行界面執行caffe,Bekeley caffe命令行界面

如何確保caffe知道如何找到爲您的圖層編寫類的文件?你只需將.py文件放在與train.prototxt相同的目錄下?

如果你寫在你需要使用python封裝接口蟒蛇的自定義類?

+0

非常感謝!我會嘗試一下並發表評論。 –

+0

請參閱[本答案](https://stackoverflow.com/a/41481539/1714410)瞭解更多關於Python圖層的信息。 – Shai

回答

2

你的Python層在prototxt兩個參數:layer:此處定義Python類名執行圖層,並moduule:這裏您可以定義層類實現的.py文件名。
當你運行caffe(無論是從命令行還是通過python界面)你都需要確保你的模塊在PYTHONPATH

+0

真棒。非常感謝。 –