一、整体流程
修改src/caffe/proto/caffe.proto文件->在caffe/include/caffe/layers与caffe/src/caffe/layers添加对应层级的头文件与源文件->编译caffe库
二、实现细节
第一步:在对应src/caffe/proto/caffe.proto文件找到message LayerParameter{…}的代码,并在其中添加
optional YOUR_LAYERS_NAME layers_param = 任意proto中未出现的数字;
[注意新增加行中的ID数值,可以为其他值(如200,500等),但是不能与当前已经存在的层参数ID相同。在Note中已经提示下一个可用的参数ID为147等。]
第二步:添加message AllPassParameter { },指定参数名称
message YOUR_LAYERS_Parameter {
optional float key = 1 [default = 0];
}
第三步:将对应在caffe/include/caffe/layers与caffe/src/caffe/layers添加对应层级的头文件与源文件->编译caffe库