写在前面:
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中,需要测试通过。