1. 程式人生 > >caffe中加入python layer的方法

caffe中加入python layer的方法

小魚趁這個機會,學學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並呼叫成功