【caffe添加新层】通道复制层ChCopy

该层的作用是将n*1*h*w维度的数据复制成n*copynum*h*w维的数据。

1.首先是.hpp和.cpp文件的编写,源码如下

《.hpp文件》

.hpp

《.cpp文件》

层setup以及reshape部分


前向传播部分


反向传播及“注册”部分

《caffe.proto文件修改》:在message中声明chcopy参数,以及重用该参数

进入caffe根目录,

make clean

make all

make all -j8

2.使用时,写好相应的prototxt文件,即可

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容