假设新增加的层命名为: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下增加对应的注册的代码。
caffe中增加自己的layer
最新推荐文章于 2018-11-22 09:42:14 发布