下载 faster-rcnn 源码

从 github 上下载支持 pytorch-1.0 的 faster-rcnn 版本

git clone -b pytorch-1.0 git@github.com:jwyang/faster-rcnn.pytorch.git FRCN-ROOT

按照要求准备好数据和与训练模型，使用的 python 版本为 3.6

cd FRCN-ROOT/lib
make

训练

然后执行训练命令时，遇到两个错误

第一个错误

ImportError: cannot import name '_mask'

先从 github 上下载最新的 cocoapi

git clone git@github.com:cocodataset/cocoapi.git

在 cocoapi/PythonAPI 下执行 make

cd cocoapi/PythonAPI
make

然后把 FRCN-ROOT/lib 下的 pycocotools 文件夹替换为 cocoapi/PythonAPI 下的这个

rm -rf FRCN-ROOT/lib/pycocotools
cp -r cocoapi/PythonAPI/pycocotools FRCN-ROOT/lib/

第二个错误

ImportError: cannot import name 'imread’

原因是 scipy 版本过高，降低版本就好了

pip install scipy==1.2.1

训练命令