在caffe中添加或修改层

在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中