(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&