mobileV3-SSD訓練自己的數據集

1.製作VOC數據集

這是最關鍵的一步,首先類別名稱一定不能亂起名,我的類別名因爲中間有空格,導致後來出錯.至於圖片名和xml的名字是否該成000001.的格式,改不改都行.

生成的XML文件中的類別名要和自己的類別名相對應.

此外 要把voc_dataset.py中的class_names改爲自己的類別名.

 

如果遇到IndexError: too many indices for array的錯誤

数据集制作参考：https://www.cnblogs.com/blog4ljy/p/9195752.html

參考https://github.com/amdegroot/ssd.pytorch/issues/224