源码链接:https://github.com/amdegroot/ssd.pytorch
首先下载源码和权重到本地,然后打开_ini_.py注释掉第二行的coco,由于我测试使用的是VOC2007数据集,不注释掉的话会报错
from .voc0712 import VOCDetection, VOCAnnotationTransform, VOC_CLASSES, VOC_ROOT
#from .coco import COCODetection, COCOAnnotationTransform, COCO_CLASSES, COCO_ROOT, get_label_map
接下来是修改voc0712.py里27、28行使得路径变为绝对路径(win10下不是绝对路径容易出错),里面有代码可以直接定位到用户的位置,所以在后面加上Desktop等就可以组成绝对路径。
# note: if you used our download scripts, this should be right
VOC_ROOT = osp.join(HOME, "Desktop/data/VOCdevkit/")
由于笔记本只有cpu,想要使用cpu来运行就需要把eval.py里427行改成4如下行所示。并且parse参数中的cuda默认值改为False
#net.load_state_dict(torch.load(args.trained_model))
net.load_state_dict(torch.load(args.trained_model,map_location=torch.device('cpu')))
parser.add_argument('--cuda'