Caffe Система глубокого обучения конвертирует Normalize в BatchNorm - PullRequest
Новая
       51

Caffe Система глубокого обучения конвертирует Normalize в BatchNorm

0 голосов
/ 11 марта 2020

Я использую 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
        }
      }
    }
...
...