用SSD-Pytorch训练自己的数据集

第一步：在Github上clone SSD-Pytorch：https://github.com/amdegroot/ssd.pytorch，根据教程一步步部署。

第二步：config.py里面的voc的classes数值为你的类别数+1，默认模仿voc进行训练；修改voc0712.py中的VOC_CLASSES和VOC_ROOT设置为自己的类别名称与路径，另外需要修改image_sets设置路径；修改multibox_loss.py中的94与95行互换位置，114行修改为       

N = num_pos.data.sum().double()   

 loss_l = loss_l.double()

loss_c = loss_c.double()

若你的pytorch版本为0.4.1则修改完毕，直接运行train.py。若你的pytorch版本为1.0.1则还需要修改train.py中的step3: change the line188,189,193,196:
loss_l.data[0] >> loss_l.data
loss_c.data[0] >> loss_c.data
loss.data[0] >> loss.data