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并调用成功