caffe 中添加新的 layer 和相应测试的过程
添加新 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...
原创
2018-11-22 09:42:14 ·
597 阅读 ·
0 评论