【Caffe实践】 添加自己的网络层

写在前面:

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中,需要测试通过。

评论 17
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值