pytorch版faster rcnn配置运行教程

╰+哭是因爲堅強的太久メ 2023-07-17 02:59 16阅读 0赞

运行环境：ubuntu18.04+CUDA10.1+python3.6+Pytorch1.2

项目地址：

[https://github.com/jwyang/faster-rcnn.pytorch][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.git

2.创建数据集软链接

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/VOCdevkit2007

3.编译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_jwyang_faster-rcnn.pytorch_issues_410]  
[https://github.com/cocodataset/cocoapi/issues/59\#issuecomment-469859646][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

【参考】

[使用faster-rcnn.pytorch训练自己数据集（完整版）][faster-rcnn.pytorch]

[记pytorch版faster rcnn配置运行中的一些坑][pytorch_faster rcnn]

[https_github.com_jwyang_faster-rcnn.pytorch]: https://github.com/jwyang/faster-rcnn.pytorch
[https_github.com_jwyang_faster-rcnn.pytorch_issues_410]: https://github.com/jwyang/faster-rcnn.pytorch/issues/410
[https_github.com_cocodataset_cocoapi_issues_59_issuecomment-469859646]: https://github.com/cocodataset/cocoapi/issues/59#issuecomment-469859646
[faster-rcnn.pytorch]: https://www.cnblogs.com/wind-chaser/p/11359521.html
[pytorch_faster rcnn]: https://www.cnblogs.com/FZfangzheng/p/10852141.html