使用SSD mobilenet训练自己的数据集

1、首先配置好环境，我是用的的torch1.7.1，和torchvision0.8.2版本，可以直接根据作者给出的requirements.txt文件进行其他安装包的配置

2、下载代码lufficc-SSD
PS:我是根据某乎上面一个博主进行代码调试的，这里给出链接SSD训练自己数据

3、先放置数据集，按照这种方式放置的话，代码里面其他位置都不需要改变了
看另外一个博主说的其实VOC2012里面的内容就是VOC2007直接复制过去的，不过我是在训练之后才看到他的博文，所以我是将自己的数据集一大半放VOC2007，小部分放的VOC2012的。annotations中放置Pascal voc类型的xml文件

4、打开ssd/config/defaults.py文件，以及configs文件夹下你需要的yaml文件，在这两个里面更改num_classes，注意这里要改成自己的数据类别数+1！！！！这个1代表背景

5、打开ssd/data/datasets/，在下面找到coco.py或者voc.py，将里面的列别改成自己的类别。知乎那个博主说的加载预训练模型的问题我没有遇到，因为我没有使用预训练模型，在defaults.py文件中将加载预训练模型设为False即可。
voc.py的69行class_name =