caffe添加自己新层

caffe增加Layer的步骤如下:

1.在./src/caffe/proto/caffe.proto增加对应的LayerParameter message,注意ID不能重复。

2.在./include/caffe/layers/下增加对应layer的声明,编写hpp文件(仿照着其它层进行编写)。

3.在./src/caffe/layers/路径下增加对应layer的cu、cpp文件。

4.在./src/caffe/gtest/中添加新层的测试代码,保证不会出错。

5.重新编译
make clean
make all -j8
make pycaffe -j8

你可能感兴趣的:(caffe学习)