caffe添加自己的层
Posted simple_wxl
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了caffe添加自己的层相关的知识，希望对你有一定的参考价值。
首先修改src/caffe/proto/下的caffe.proto,修改好后需要编译
然后修改include/caffe/layers/logwxl_layer.hpp
然后修改src/caffe/layers/logwxl_layer.cpp和logwxl_layer.cu
最后make all ,make test ,make runtest
在make runtest的时候，卧槽有一个错误，好像是没有updateV1parameter什么的，但是不影响最后的训练
以上是关于caffe添加自己的层的主要内容，如果未能解决你的问题，请参考以下文章