caffe win添加新层

1、编写.h和.cpp .cu文件

将.hpp文件放到路径caffe-windows\caffe-master\include\caffe\layers下 
将.cpp文件和.cu放到路径caffe-windows\caffe-master\src\caffe\layers下

修改caffe-windows\caffe-master\src\caffe\proto\caffe.proto,需要修改两处
1.找到 LayerParameter 描述,增加一项:

message LayerParameter {
...
optional ShuffleChannelParameter shuffle_channel_param = ;
} 注意,这里的参数设置不能与LayerParameter中其他参数相同

2.仍然在 caffe.proto 中,增加

ShuffleChannelParameter 

声明,位置任意。其功能是可以用于从 prototxt 中读取预设值。

message ShuffleChannelParameter {
optional uint32 group = [default = ]; // The number of group
}

将我们定义的层加载到Caffe中去

在libcaffe项目中include文件夹 layers——添加.h文件

在libcaffe项目中src文件夹layers——添加.cpp文件

重新编译caffe即可

上一篇:集合之LinkedList


下一篇:golang基础教程——字符串篇