用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