[人工智能]（14）目标检测_SSD训练代码基于pytorch搭建代码

1、主要参数

（1）anchor有两种，如果小目标

anchors_size = [21, 45, 99, 153, 207, 261, 315]

如果一般目标

anchors_size    = [30, 60, 111, 162, 213, 264, 315]

（2）采用预训练模型，使用adam

前50步冻结训练，50步开始不冻结训练

（二）从主干网络的预训练权重开始训练：
    #       Adam：
    #           Init_Epoch = 0，Freeze_Epoch = 50，UnFreeze_Epoch = 100，Freeze_Train = True，optimizer_type = 'adam'，Init_lr = 6e-4，weight_decay = 0。（冻结）
    #           Init_Epoch = 0，UnFreeze_Epoch = 100，Freeze_Train = False，optimizer_type = 'adam'，Init_lr = 6e-4，weight_decay = 0。（不冻结）

其中Freeze_Epoch为模型冻结训练的Freeze_Epoch，UnFreeze_Epoch为模型总共训练的epoch

（3）设置了pretrained=true&

开通VIP 解锁文章