如何在新版本caffe里添加新的一层

1、首先写好你添加层的cpp和cu文件,另外在cpp文件的最后写上REGISTER_LAYER_CLASS(LAYERTYPENAME);其中LAYERTYPENAME是你定义的layer的类名，如MyInputLayer，那么就是REGISTER_LAYER_CLASS(MyInput);，注意不是MyInputLayer。

2、然后在相关的头文件里添加层类的声明

3、在proto文件里定义层的类型和message

4、在upgrade_proto.cpp里const char* UpgradeV1LayerType(const V1LayerParameter_LayerType type) {
  switch (type) {}里添加相关代码case...return

5、写prototxt网络模型文件的时候要注意type的格式