一般说的BN操作是指caffe中的BatchNorm+Scale,
要注意其中的use_global_states:默认是true【在src/caffe/caffe.proto】
训练时:use_global_states:false
测试时:use_global_states:true
【重要】可以看到很多都是如下:
layer { bottom: "conv1_1" top: "conv1_1" name: "bn_conv1_1" type: "BatchNorm" batch_norm_param { } }
这样就是在代码执行过程中,会自动选择训练时用false,test时用true
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:caffe中BN层 - Python技术站