在caffe中添加自定义层

Caffe中添加自定义层[6,7]

1）头文件

路径：py-faster-rcnn/caffe-fast-rcnn/include/caffe/layers

2）CPP或者CU文件

fasterrcnn_huawei/py-faster-rcnn/caffe-fast-rcnn/src/caffe/layers

3）caffe.proto定义该层的参数

py-faster-rcnn/caffe-fast-rcnn/src/caffe/proto/caffe.proto

proposal_layer.cpp

3.1）message LayerParameter { } 修改

添加消息函数的ID，设置ID。

message LayerParameter {

optional WindowDataParameter window_data_param = 129; // 目前，唯一的值，不能重复，最大值为143以及另外较大数值

optional ProposalParameter proposal_param = 144;

}

3.2) 添加消息函数（任意位置）

按顺序指明参数。

message ProposalParameter {

optional float diff_scale = 1 [default = 1]; //默认梯度不缩放， 暂时不添加

}

3.3) 添加Layer ID（可能不需要添加）

首先，在 message V1LayerParameter { }中的

enum LayerType {

PROPOSAL=xxx; // 不重复即可

}

然后在继续添加

message V1LayerParameter {

enum LayerType {

WINDOW_DATA = 24;

PROPOSAL=xxx; // 不重复即可

}

optional WindowDataParameter window_data_param = 20

}

3.4) 添加参数定义message V0LayerParameter {}（可能不需要添加）

不再推荐的方式，这个与ProposalParameter 是什么关系？？

message V0LayerParameter {

}

3.5 定义一个层

layer{

name: "proposal"

type: "Proposal"

bottom: "" // 与python版本一致

top: "" // 与python版本一致

}

注：可以参考DataLabelParameter

4）注册层

在CPP文件中添加

INSTANTIATE_CLASS(DiffCutoffLayer);

REGISTER_LAYER_CLASS(DiffCutoff);

小结：定义一个新的层，简单来说分为两个步骤：

参考资料[3,4,5,6,712]

1）声明和定义该层的类，分别在hpp和cpp文件中声明和定义该类，并且需要在cpp文件中对该层进行注册，用下面的两行代码（下面所示）注册后，则不需要在layer_factory.cpp文件中

添加构造该新层对象的函数。通常新层会继承子已有的层次，例如layer.cpp层，需要在新层中定义基类中的纯虚函数。例如对于layer层，包括三个虚函数（Backward_cpu, Forward_gpu和Reshape）。

对于只包含测试过程的层，可以通过在Backward_cpu的定义代码中只添加“NOT_IMPLEMENTED;”来表示未定义反向传播函数。

INSTANTIATE_CLASS(DiffCutoffLayer);

REGISTER_LAYER_CLASS(DiffCutoff);

2）配置该层：包括注册和在caffe.proto中设置相关参数。在较新版本的caffe中，用户只需要

在编译caffe的过程中，会根据caffe.proto文件中定义的参数接口来编译相关层的代码。一般只需要做3.1和3.2两部即可。

注意：不需要在layer_factory.cpp文件中添加新层的头文件。

[1] http://zhangliliang.com/2015/05/17/paper-note-fast-rcnn/

[2] http://blog.csdn.net/anxiaoxi45/article/details/46522155

[3] caffe/src/caffe/proto/caffe.proto

[4] https://developers.google.com/protocol-buffers/

[5] http://caffe.berkeleyvision.org/tutorial/layers.html

[6] http://caffecn.cn/?/question/388

[7] http://blog.csdn.net/shuzfan/article/details/51322976

[8] http://blog.csdn.net/seven_first/article/category/5721883 // caffe源码解析

[9] https://github.com/shicai/Caffe_Manual // 模型建立过程

[10] https://github.com/BVLC/caffe/wiki/Solver-Prototxt

[11] http://karpathy.github.io/neuralnets/

[12] https://github.com/BVLC/caffe/wiki/Development#developing-new-layers