MMDetection安装过程记录

记录自本人安装过程,环境建议:

  • Linux or macOS (Windows is not currently officially supported)
  • Python 3.6+
  • PyTorch 1.3+
  • CUDA 9.2+ (If you build PyTorch from source, CUDA 9.0 is also compatible)
  • GCC 5+

1. torch & torchvision

我使用的是 torch 1.5.1 + torchvision 0.6.1

pip install torch==1.5.1 torchvision==0.6.1 [-i https://pypi.douban.com/simple]

https://pytorch.org/get-started/previous-versions/

2. mmcv

pip install mmcv-full

3. mmdetection

git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection

pip install -r requirements/build.txt
pip install -v -e .  # or "python setup.py develop"

4. test

下载数据

mkdir checkpoints
wget -c https://open-mmlab.s3.ap-northeast-2.amazonaws.com/mmdetection/v2.0/mask_rcnn/mask_rcnn_r50_caffe_fpn_mstrain-poly_3x_coco/mask_rcnn_r50_caffe_fpn_mstrain-poly_3x_coco_bbox_mAP-0.408__segm_mAP-0.37_20200504_163245-42aa3d00.pth \
      -O checkpoints/mask_rcnn_r50_caffe_fpn_mstrain-poly_3x_coco_bbox_mAP-0.408__segm_mAP-0.37_20200504_163245-42aa3d00.pth

此时查看文件目录树如下

> tree checkpoints
checkpoints
└── mask_rcnn_r50_caffe_fpn_mstrain-poly_3x_coco_bbox_mAP-0.408__segm_mAP-0.37_20200504_163245-42aa3d00.pth

运行image_demo.py测试mmdet是否安装成功

python demo/image_demo.py demo/demo.jpg configs/mask_rcnn/mask_rcnn_r50_caffe_fpn_mstrain-poly_3x_coco.py checkpoints/mask_rcnn_r50_caffe_fpn_mstrain-poly_3x_coco_bbox_mAP-0.408__segm_mAP-0.37_20200504_163245-42aa3d00.pth 

若安装成功,会显示如下图像:

image-20200925142629606
image-20200925142629606

Some ERROR

libcudart.so.10.1: cannot open shared object file: No such file or directory

先卸载原有cudatoolkitconda uninstall cudatoolkit

再安装10.1版本:conda install cudatoolkit=10.1

Licensed under CC BY-NC-SA 4.0
comments powered by Disqus