如果您使用的是自定義python圖層 - 並且假設您在python中正確編寫了該類 - 假設該類的名稱爲"my_ugly_custom_layer"
;並在linux命令行界面執行caffe,Bekeley caffe命令行界面
如何確保caffe知道如何找到爲您的圖層編寫類的文件?你只需將.py文件放在與train.prototxt相同的目錄下?
或
如果你寫在你需要使用python封裝接口蟒蛇的自定義類?
如果您使用的是自定義python圖層 - 並且假設您在python中正確編寫了該類 - 假設該類的名稱爲"my_ugly_custom_layer"
;並在linux命令行界面執行caffe,Bekeley caffe命令行界面
如何確保caffe知道如何找到爲您的圖層編寫類的文件?你只需將.py文件放在與train.prototxt相同的目錄下?
或
如果你寫在你需要使用python封裝接口蟒蛇的自定義類?
你的Python層在prototxt兩個參數:layer:
此處定義Python類名執行圖層,並moduule:
這裏您可以定義層類實現的.py文件名。
當你運行caffe
(無論是從命令行還是通過python界面)你都需要確保你的模塊在PYTHONPATH
真棒。非常感謝。 –
非常感謝!我會嘗試一下並發表評論。 –
請參閱[本答案](https://stackoverflow.com/a/41481539/1714410)瞭解更多關於Python圖層的信息。 – Shai