手把手教你使用SSD(新手必看)
下载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,
用户评论