自定義 Python 圖層
- 使用 Python 層構建 Caffe
Caffe 需要使用 WITH_PYTHON_LAYER
選項進行編譯:
WITH_PYTHON_LAYER=1 make && make pycaffe
- 我應該在哪裡儲存類檔案?
你有兩個選擇(至少我知道)。你可以將自定義圖層檔案儲存在與執行 caffe 命令相同的資料夾中(可能是你的原型檔案檔案所在的位置)。另一種方式,也是我最喜歡的方法,是將所有自定義圖層儲存在資料夾中,並將此資料夾新增到 PYTHONPATH 中。