码迷,mamicode.com
首页 > 其他好文 > 详细

Caffe 中添加自己的网络层

时间:2015-06-09 23:46:15      阅读:1007      评论:0      收藏:0      [点我收藏+]

标签:

写在前面:

Caffe 中有众多的网络层,最新版本的代码已经涵盖了很多种类型的网络层,然而,有时候由于各种原因,其给定的网络层不能满足我们的要求,这时候就要对其更改,以使其满足自己的需求,感谢作者开源代码以及众多的代码维护者。

由于Caffe 中的网络层都是直接或者间接地给予Layer 基类,所以,在我们需要添加新的类型时,就需要选择好自己的基类,以使我们能够更好的利用基类已有的一些方法。我们新建的类可以基于
1. 直接继承于Layer
2. 继承于DataLayer
3. 继承于NeuronLayer
4. 继承于LossLayer
5. 或者如果是直接想改进某个层的代码,可以直接继承该类,并做相对应的修改即可。

具体步骤:

第一步, 添加头文件.h ,可以自己新建一个头文件,或者追加到一个已有的头文件后面,比如include/DataLayer.hpp 等, 在里面定义一些类变量等。

第二步,添加实现文件,创建类的实现函数,放在src/XXX.cpp 中,如果有cuda实现,则需要添加相对应的.cu文件实现,主要需要写LayerSetup ,reshape ,forward ,backup 函数。

第三步, 注册类,到src/caffe/layer_factory.cpp 中添加对应的类,在其中加入

REGISTER_LAYER_CLASS(MYLAYER, MyLayer);

第四步,在src/caffe/proto/caffe.proto 文件中,添加对应的类以及类参数。
在enum LayerType {}中添加一行

enum LayerType {
MYLAYER = ID 
}
//其中MYLAYER 是新添加的类名,ID是类编号,注意不能更之前的重复。

再添加类参数

optional MylayerParameter mylayer_param = ID2;
//其中MylayerParameter 是网络层参数,ID2 是ID

最后定义类参数

message MylayerParameter {
  optional uint32 myparam1 = 1; 
  optional bool myparam2 = 2 [default = true]; 
}

第五步,写测试文件,这个是可选的,不过为了合并到github中,需要测试通过。

Caffe 中添加自己的网络层

标签:

原文地址:http://blog.csdn.net/chenriwei2/article/details/46432727

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!