1、编写.h和.cpp .cu文件
将.hpp文件放到路径caffe-windowscaffe-masterincludecaffelayers下
将.cpp文件和.cu放到路径caffe-windowscaffe-mastersrccaffelayers下
修改caffe-windowscaffe-mastersrccaffeprotocaffe.proto,需要修改两处
1.找到 LayerParameter 描述,增加一项:
message LayerParameter { ... optional ShuffleChannelParameter shuffle_channel_param = 164; }
注意,这里的参数设置不能与LayerParameter中其他参数相同
2.仍然在 caffe.proto 中,增加
ShuffleChannelParameter
声明,位置任意。其功能是可以用于从 prototxt 中读取预设值。
message ShuffleChannelParameter { optional uint32 group = 1[default = 1]; // The number of group }
将我们定义的层加载到Caffe中去
在libcaffe项目中include文件夹 layers——添加.h文件
在libcaffe项目中src文件夹layers——添加.cpp文件
重新编译caffe即可