caffe python layer初次使用记录

书写方式举例
layer{
name: "Leaky25"
type: "Python"
bottom: "connect1"
top: "Leaky25"
python_param{
module: "mylayer"#ｐｙｔｈｏｎ的文件名字为mylayer.py
layer: "LeakyLayer"#层的名称为LeakyLayer
}
}

层的实现：
import caffe
import numpy as np
import yaml

class LeakyLayer(caffe.Layer):
def setup(self, bottom, top):
pass
def reshape(self, bottom, top):
print "reshape"
top[0].reshape(*(bottom[0].shape))

def forward(self, bottom, top):
print "forward"
count = bottom[0].count
data_bot = bottom[0].data
data_top = top[0].data
shape = data_bot.shape
for i in xrange(shape[0]):
for j in xrange(shape[1]):
if data_bot[i][j] < 0:
data_top[i][j] = data_bot[i][j]*0.1
else:
data_top[i][j] = data_bot[i][j]

def backward(self, top, propagate_down, bottom):
pass