MXNet-SSD 再识--Mobilenet_v2

1。加载预训练模型

除了MXNet-SSD 初识中介绍的参数外，在此还需要再添加另一个参数：

--pretrained

2。修改/home/whut/yyCode/ssd/symbol/symbol_factory.py

elif network=='mobilenet_v2':
        image_shape = '3,224,224'
        network = 'mobilenet_v2'
        from_layers = ['relu6_1_expand', 'relu6_4', '', '', '', '']
        num_filters = [-1, -1, 512, 256, 256, 128]
        strides = [-1, -1, 2, 2, 2, 2]
        pads = [-1, -1, 1, 1, 1, 1]
        sizes = [[.1, .141], [.2, .272], [.37, .447], [.54, .619], [.71, .79], [.88, .961]]
        ratios = [[1, 2, .5], [1, 2, .5, 3, 1. / 3], [1, 2, .5, 3, 1. / 3], [1, 2, .5, 3, 1. / 3], \
                  [1, 2, .5], [1, 2, .5]]
        normalizations = -1
        steps = []
        return locals()

添加mobilenet_v2的相关说明。

3。开始训练

尝试一：batch-size=128，lr=0.005，--lr-steps='120,240'，--end-epoch=360

 发现：过早的调整了学习率。

尝试二：batch-size=128，lr=0.005，--lr-steps='150,280'，--end-epoch=360