1.创建conda虚拟环境
conda create -n open-mmlab python=3.7 -y
conda activate open-mmlab
2.安装pytorch 和 torchvision
#国内网络环境,先替换conda源,换源后直接:
conda install pytorch torchvision -y #不加-y会弹出是否同意安装(y/n),加上默认同意
#境外网络环境可直接执行:
conda install pytorch torchvision -c pytorch
3.安装mmcv
pip install mmcv-full==1.0.4 #最新版是1.0.5,但该版本安装后运行报错,原因未知,降级可解决
4.安装cocoapi
国内服务器连接git网络条件较差,遂先用代理下载至本地再上传到服务器上用source安装
https://github.com/cocodataset/cocoapi/tree/master/PythonAPI
python setup.py build
python setup.py install
---Update---
实战中由于cocoapi官方维护已经落后,可能会报错:
AttributeError: 'COCO' object has no attribute 'get_cat_ids'
该问题比较普遍,详见https://github.com/open-mmlab/mmdetection/issues/2913
解决方案:
下载open-mmlab优化的pycocotools覆盖安装
https://github.com/open-mmlab/cocoapi/tree/master/pycocotools
下载上传后,直接
python setup.py install
5.配置mmdet
国内服务器连接git网络条件较差,遂也是先用代理下载至本地再上传到服务器上用source配置
https://github.com/open-mmlab/mmdetection/blob/master/docs/install.md
cd mmdetection/
python setup.py develop