添加新 layer
添加一个简单的 layer 可以参考 https://github.com/BVLC/caffe/wiki/Simple-Example:-Sin-Layer 。
几个主要的过程如下:
- 添加
include/caffe/layers/your_layer.hpp
- 添加
src/caffe/layers/your_layer.cpp
- 添加
test/test_your_layer.cpp
- 回到 build 文件夹
cmake -DCPU_ONLY=1 .. && make -j 4 && make test && make runtest && test/test.testbin --gtest_filter='*YouTestrClassName*'
参考
https://github.com/BVLC/caffe/wiki/Simple-Example:-Sin-Layer
https://github.com/BVLC/caffe/wiki/Development