Advertisement

配置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的博客-博客]( "pytorch 1.0实现faster R-CNN,检验自己的数据集_Soul·Top的博客-博客")

不过呢,就不用先下载再切换分支了,直接在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的博客-博客]( "安装faster-rcnn pytorch1.0.0版本以及importerror cannot import name '_mask'_hecaozhong的博客-博客")

就是下载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的博客-博客]( "TypeError: load\(\) missing 1 required positional argument: ‘Loader‘_one_wangtester的博客-博客")

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

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

即可。

全部评论 (0)

还没有任何评论哟~