在caffe中添加层主要有下面几个步骤:
修改proto文件
文件位置在caffe/src/caffe/proto
在这个文件中添加或修改层名和层的参数
注意数字索引不能重复
message BlobProto {
optional BlobShape shape = 7;
repeated float data = 5 [packed = true];
repeated float diff = 6 [packed = true];
repeated double double_data = 8 [packed = true];
repeated double double_diff = 9 [packed = true];
// 4D dimensions -- deprecated. Use "shape" instead.
optional int32 num = 1 [default = 0];
optional int32 channels = 2 [default = 0];
optional int32 height = 3 [default = 0];
optional int32 width = 4 [default = 0];
}
修改对应的.cpp和.h文件
位置在caffe/src/caffe/layers
和caffe/include/caffe/layers
中