自定義 Python 圖層

- 使用 Python 層構建 Caffe

Caffe 需要使用 WITH_PYTHON_LAYER 選項進行編譯:

WITH_PYTHON_LAYER=1 make && make pycaffe

- 我應該在哪裡儲存類檔案?

你有兩個選擇(至少我知道)。你可以將自定義圖層檔案儲存在與執行 caffe 命令相同的資料夾中(可能是你的原型檔案檔案所在的位置)。另一種方式,也是我最喜歡的方法,是將所有自定義圖層儲存在資料夾中,並將此資料夾新增到 PYTHONPATH 中。

參考

  1. 克里斯托弗佈雷斯的部落格
  2. Caffe Github
  3. 堆疊溢位