在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))
即可。