win10+cpu+SSD-pytorch运行记录

源码链接: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'
评论 7
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值