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下增加對應的註冊的代碼。


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