配置faster-rcnn pytorch1.0.0版本实现faster R-CNN

GitHub - jwyang/faster-rcnn.pytorch at pytorch-1.0下载项目。

(切记:一定是1.0分支下,不是master主分支下,因为只有1.0分支下才支持pytorch-1.0)

可大致参考:pytorch 1.0实现faster R-CNN,检验自己的数据集_Soul·Top的博客-CSDN博客

不过呢,就不用先下载再切换分支了,直接在1.0分支下下载就行。

注意两个问题:

(1)若按照官方Github给出的步骤在lib文件夹下执行

python setup.py build develop

之后运行python demo.py出现cannot import name '_mask'的问题的话,

是因为编译版本不匹配,应该是默认python2.7的原因,就是如果自己的python版本是2.7则应该不会出现该问题,官方也写了:

The default version is compiled with Python 2.7, please compile by yourself if you are using a different python version.

 解决该问题的话,可以参考如下博客中步骤2:安装faster-rcnn pytorch1.0.0版本以及importerror cannot import name '_mask'_hecaozhong的博客-CSDN博客

就是下载cocoapi,在cocoapi/PythonAPI中重新make,然后将faster-rcnn.pytorch/lib里面的pycocotools文件夹删除,替换为PythonAPI中刚编译成功的pycocotools文件夹即可,该解决方法也可重点参考:no module named _mask? · Issue #59 · cocodataset/cocoapi · GitHub

 

(2)若运行python demo.py时出现以下问题:

TypeError: load() missing 1 required positional argument: 'Loader'

则是因为YAML 5.1版本的原因,可参考:TypeError: load() missing 1 required positional argument: ‘Loader‘_one_wangtester的博客-CSDN博客

就是将相应代码中:yaml_cfg = edict(yaml.load(f))

修改为:yaml_cfg = edict(yaml.load(f, Loader=yaml.FullLoader))

即可。

  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值