mobilenetv3-ssd测试及训练

项目地址:https://github.com/shaoshengsong/MobileNetV3-SSD

需要注意的几个问题:

1.在测试一张图片的时候会报错:

expected backend CPU and dtype Float but got backend CUDA and dtype Float

解决方法:

If you are trying to use GPU, change device in create_mobilenetv3_ssd_lite_predictor to torch.device('cuda:0'), it works for me.


2.首次训练时:
如果--dataset_type选择open_images则需要事先运行

python open_images_downloader.py --root ~/data/open_images --class_names "Handgun,Shotgun" --num_workers 20进行下载.

默认使用的是VOC模式,所以如果下载VOC数据集训练则:

python train_ssd.py --datasets ~/data/VOC2007/ --validation_dataset ~/data/VOC2007/ --net mb1-ssd 

--net mb3-ssd-lite --scheduler cosine --lr 0.01 --t_max 100 --validation_epochs 5 --num_epochs 100 --base_net_lr 0.001 --batch_size 5
一定要加--validation_dataset 由于我只下载了VOC2007 所以验证集也用了VOC2007 注意在　VOC2007/imageSets/Main/下新建test.txt不然会找不到
更多细节参考:https://github.com/qfgaohao/pytorch-ssd

運行python run_ssd_example.py報錯:expected device cpu but got device cuda:0
解決:https://github.com/shaoshengsong/MobileNetV3-SSD/issues/5