下载SSD代码:
https://github.com/amdegroot/ssd.pytorch
将下载的代码解压后在data文件夹下新建VOCdevkit文件夹,然后将VOC2007数据集复制到该文件夹下面。
下载权重文件放在weights文件夹下面。下载地址:
https://s3.amazonaws.com/amdegroot-models/vgg16_reducedfc.pth
修改config.py代码
# SSD300 CONFIGS
voc = {
'num_classes': 3,//将类别改为:类别+1(背景)
'lr_steps': (80000,
1