Frcnn Pytorch1.0 Linux调试
FasterRCNN PyTorch
最后更新 2020-05-08 17:36 阅读 7034
最后更新 2020-05-08 17:36
阅读 7034
FasterRCNN PyTorch
代码链接:https://github.com/jwyang/faster-rcnn.pytorch/tree/pytorch-1.0
pytorch 1.0 cuda 8.0 python 3.6 gcc 5.2(一定要4.8以上……)
1.python setup.py build develop (这个是因为我下错了 下的0.4的版本 如果是pytorch1.0版本跑0.4可以这么修改) 参考答案 将setup.py 中的 from distutils.core import setup #替换为 from setuptools import setup 123 重新bulid就好了
2、ImportError: cannot import name ‘_mask’ 这个是因为没有安装cocoapi 根据issue可以安装
cd data git clone https://github.com/cocodataset/cocoapi.git cd coco/PythonAPI make
如果make之后还是没有成功,那就删除lib下面pycocotools的一些东西,然后重新make。 一定要在make之后再setup一遍。 主要看输出如果有skip任何文件,建议删除缓存,再进行setup。
3、Segmentation fault loading …….pth之后出现的错误。 根据issue 知道是gcc版本过低(我的环境是4.7,这个代码最好使用5.2) 安装gcc废了我好大的力气…… 如果你有梯子,那这个方式适合你
if you use anaconda, you can smplily find the gcc-5-5.2.0 on the Anaconda Clould. Install the gcc follow the command, after that the gcc version will change to 5.2.0.
而我属于不可以的那种,conda install半天……下了一半终止了,因为有缓存包,所以conda 也开始报错,报错内容是segentation fault,升级conda也不能用,这个时候可以使用
conda clear -t #清除未缓存完成的包 conda clear -a #清除所有的
具体参考指令说明 好了,现在在线安装不行了,改为离线吧。 离线gcc 5.2 Linux版链接 下载好了 pip install就好了 在你的环境里 gcc -v查看版本 然后!还要删除缓存,make以及setup !!
4、RuntimeError: Not implemented on the CPU (ROIAlign_backward at c:\users\lenovo\faster-rcnn.pytorch\faster-rcnn.pytorch\lib\model\csrc\ROIAlign.h:44) (no backtrace available) 这个根据issue 参数放在py文件里面修改 指令只写 python trainval_net.py --cuda就好了…… 真是神奇……
5、题外话,想要本地调试服务器代码,需要使用pycharm专业版……
找官网下就好了…