Caffe深度学习框架将Normalize转换为BatchNorm

时间:2020-03-11 13:34:38

标签: deep-learning caffe

我正在使用TIDL(TI深度学习库)转换深度学习模型以用于嵌入式系统。 我使用的版本不支持Caffe的“ Normalize”层,因此我想以某种方式使用“ BatchNorm”。通过直接编辑prototxt文件是否可以实现类似的功能?如果是这样,我如何将Normalize修改为BatchNorm,因为它们采用了不同的参数。

非常感谢您的帮助。谢谢。

...
    layer {
      name: "conv10_2_relu"
      type: "ReLU"
      bottom: "conv10_2"
      top: "conv10_2"
    }
    layer {
      name: "conv4_3_norm"
      type: Normalize
      bottom: "conv4_3"
      top: "conv4_3_norm"
      norm_param {
        across_spatial: false
        scale_filler {
          type: "constant"
          value: 20
        }
        channel_shared: false
      }
    }
    layer {
      name: "conv4_3_norm_mbox_loc"
      type: "Convolution"
      bottom: "conv4_3_norm"
      top: "conv4_3_norm_mbox_loc"
      param {
        lr_mult: 1
        decay_mult: 1
      }
      param {
        lr_mult: 2
        decay_mult: 0
      }
      convolution_param {
        num_output: 16
        pad: 1
        kernel_size: 3
        stride: 1
        weight_filler {
          type: "xavier"
        }
        bias_filler {
          type: "constant"
          value: 0
        }
      }
    }
...

0 个答案:

没有答案