deep-learning - 如何将自定义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 层合并到其中。
（或者，这似乎是一个 deeplab2 Caffe 存储库，最新，支持 BN：https ://github.com/xmyqsh/deeplab-v2 ）