caffe中增加自己的layer

假设新增加的层命名为:NEW

1. 在src/proto的LayerParameter 的 LayerType下 加  NEW= 数字;

2. 在src/layer_factory.cpp中, 加 
    case LayerParameter_LayerType_NEW:

         return new NewLayer<Dtype>(param);

3.在src/layers/下 加 new_layer.cpp和 new_layer.cu代码;

4.在include/caffe/vision_layers.hpp 下增加代码(也可能在其他的.hpp下增加,如 common_layer.hpp, 
   neuron_layer.hpp等,具体视增加的layer类型决定);

5.在upgrade_proto.cpp下增加对应的注册的代码。


你可能感兴趣的:(layer,caffe,自己的)