注:刚接触R-CNN、Fast R-CNN、Faster R-CNN,就尝试用mmdetection,在coco数据集上跑Faster R-CNN,训练测试完就来记录一下这个过程的一些坑
参考
关于跑通mmdetection,建议主要看第一个网址,第二个网址参考为主,因为我按原博的步骤,跑了两次都没跑出来(可能是我太菜了)
2020.06.03更新:
这是我最近使用新版的mmdetection的时候,又重新记录了安装的过程。安装mmdetection具体可以参考该博客,但是下面的记录也是有些值得借鉴的意义的,可以互相借鉴学习。
安装教程
本人系统环境:
- Ubuntu 16.04
- Cuda 9.0 + Cudnn 7.0.5
- Python 3.7 (mmdetection要求Python版本需要3.4+)
- Anaconda 3
几个依赖库的安装:
- PyTorch 1.1(很多地方都是1.0+版本,但博主跑1.0.1的时候,不行???)
- mmcv
- cython
Anaconda3安装、环境配置
这个安装过程其实没有什么讲的,就是一直回车回车(还有yes/no的时候,多看两眼)。
但是博主在安装后,出现了几个小问题,可以参考下面两篇博客
注:因为在实验室服务器跑,时常清理,比较懒,就没有创建虚拟环境了。自行csdn一下。
安装pytorch
强烈建议去官网看具体的命令,根据自己的各个版本选择相应的命令。
博主在这安装这一步时,一直出错,在github网址上的pytorch要求的版本时1.0+。但是我在安装1.0.1的版本后,训练的时候出现了下面的问题
然后又自己试了下这个属性有没有,果真:
AttributeError: module 'torch.nn' has no attribute 'SyncBatchNorm'
torch1.0.1的版本没有SyncBatchNorm。博主找了各种解决办法,也没找到(网上关于这一块的问题解决好像没有)。又去升级了版本。然后也不行(不知道是不是我自己升级的问题)
最后就卸载重装了。
卸载的话,可能会遇到说,没有安装这个库,但是你明明有这个库啊?
pip uninstall pytorch
解决办法:
你用conda安装,就要有conda卸载
用pip安装,就用pip卸载