caffe 中添加新的 layer 和相应测试的过程

添加新 layer

添加一个简单的 layer 可以参考 https://github.com/BVLC/caffe/wiki/Simple-Example:-Sin-Layer 。
几个主要的过程如下:

  1. 添加 include/caffe/layers/your_layer.hpp
  2. 添加 src/caffe/layers/your_layer.cpp
  3. 添加 test/test_your_layer.cpp
  4. 回到 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

你可能感兴趣的:(C/C++学习,机器学习,caffe)