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的格式