pytorch版faster rcnn配置运行教程
发布时间
阅读量：
阅读量
运行环境：ubuntu18.04+CUDA10.1+python3.6+Pytorch1.2
项目地址：
https://github.com/jwyang/faster-rcnn.pytorch
按照README.md文件配置，但是有以下几点在配置过程中需要注意，建议这些都作为参考配合食用，效果更佳
1.下载代码分支
【注】 以上的项目地址是源码主分支，支持pytorch-0.4.0，但我的运行环境为Pytorch1.2，因此需要下载作者提供的pytorch-1.0分支
git clone -b pytorch-1.0 https://github.com/jwyang/faster-rcnn.pytorch.git2.创建数据集软链接
README.md文件中参考提供的方式
cd $FRCN_ROOT/data
ln -s $VOCdevkit VOCdevkit2007【注】创建软链接使用相对路径于创建时所在路径有关，按照这种方式创建后运行时提示找不到文件，所以我使用了绝对路径创建，如下
ln -s /home/zr/faster-rcnn.pytorch-pytorch-1.0/VOCdevkit /home/zr/faster-rcnn.pytorch-pytorch-1.0/data/VOCdevkit20073.编译CUDA依赖
README.md文件中步骤是
cd lib
python setup.py build develop【注】因为项目中的/lib/pycocotools里的文件过于老旧，这样编译后之后会出现ImportError: cannot import name '_mask'
训练时遇到无法导入_mask的问题，这个问题在github上也有人提问
https://github.com/jwyang/faster-rcnn.pytorch/issues/410
https://github.com/cocodataset/cocoapi/issues/59#issuecomment-469859646
看了很多后，我是这样解决的：
1.##下载cocoapi代码
git clone https://github.com/cocodataset/cocoapi.git
2.##单独编译PythonAPI
cd PythonAPI
python setup.py build_ext --inplace
3.##删除faster-rcnn.pytorch/lib/下的pycocotools，将编译后的pycocotools文件夹复制到faster-rcnn.pytorch/lib/下
4.##现在可以在faster-rcnn中按原来的方法编译了
cd lib
python setup.py build develop【参考】
全部评论 (0)
还没有任何评论哟~