如何将自定义bn图层添加到caffe

Question

我从GitHub下载了一个deeplabV2项目，发现它没有"BN"图层。
我想将它添加到caffe代码但遇到此错误：

src/caffe/layers/batch_norm_layer.cpp:250:31: 
error: macro "REGISTER_LAYER_CLASS" requires 2 arguments, but only 1 given REGISTER_LAYER_CLASS(BatchNorm);

请问如何解决？

Answer 1

看起来你正在使用一个旧的Caffe版本，它为REGISTER_LAYER_CLASS宏提供2个参数，但是BN层适用于较新的Caffe版本，其中宏被更改为只接受一个参数（这是图层类型） 您可以在其他图层中检查REGISTER_LAYER_CLASS的其他参数是什么，并将其相应地添加到您的BN图层，或者获取Caffe的更新版本并将deeplab图层合并到其中。
（或者，这似乎是最新的，具有BN支持的deeplab2 Caffe回购：https://github.com/xmyqsh/deeplab-v2）