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下增加对应的注册的代码。


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章