我正在使用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
}
}
}
...