用python定义Layer的时候, 需要实现以下几个方法:
class MyPythonLayer(caffe.Layer):
def setup(self, bottom, top):
pass
def reshape(self, bottom, top):
pass
def forward(self, bottom, top):
pass
def backword(self, top, propagate_down, bottom):
pass
各个方法都是干什么用的, 什么时候调用呢?
setup
: 初始化layer, 只调用一次.reshape
: 每次forward之前都会调用一次, 用于将blobs reshape到需要的形状. (应该)是为forward数据分配内存空间. 注意, 是先调用完所有layer的reshape方法才开始调用它们的forward方法. 不要在reshape方法里处理数据的数值, 因为reshape方法运行时, bottom传入的数据并不是forward来的数据, 确切的说, 都只是分配了空间并用0填充. 在forward之前一定要有top[i].reshape
调用, 假如shape不变的话, 可以在setup
里面只执行一次, 否则就把top[i].reshape
操作写在MyPythonLayer.reshape()
方法里forward
: forward时调用backward
: backward时调用.