caffe中加入python layer的方法
阿新 • • 發佈:2019-01-08
小魚趁這個機會,學學python layer 的編寫及實現
要想自己定義網路層的實現,如果c++程式碼夠強大,可以在caffe底層編寫cpp檔案,進行caffe的編譯
但是如果c++能力不是特別強,可以嘗試使用caffe下的python layer實現
1、 Makefile.config的
#WITH_PYTHON_LAYER := 1改為WITH_PYTHON_LAYER := 1
Caffe目錄重新編譯Caffe
make clean
make all
2、將python路徑加入到環境變數中
sudo vim ~/.bashrc
再最後新增export PYTHONPATH = /your/caffe/python/path
3、自己編寫py檔案,結構如下
import caffe
class MyLayer(caffe.Layer):
def setup(self, bottom, top):
pass
def reshape(self, bottom, top):
pass
def forward(self, bottom, top):
pass
def backward(self, top, propagate_down, bottom):
pass
可參考自己編寫py檔案
4、因為caffe是將每一個py檔案讀取為module,如果不把py檔案的路徑加入到環境變數的話,將會出現import error:no module namedXXX
解決辦法
sudo vim /usr/local/lib/python2.7/dist-packages/Mymodule.pth
建立環境變數檔案,字尾一定要為pth
在裡面加入.py檔案所在目錄即可
同時,py檔案中要加入
#!/usr/bin/env python
# -*- coding: utf-8 -*-
caffe_root = '/your/caffe/'
import sys
sys.path.insert(0, caffe_root + 'python')
import caffe
不然進入python
環境,import caffe
成功,但執行呼叫該py檔案的sh檔案會報錯說no module named caffe
這樣就實現自己寫python layer並呼叫成功