安装mmdetection

本机环境

cuda 11.2
python 3.7

操作命令

conda create -n openmmlab python=3.7 -y
conda activate openmmlab

pip install torch===1.7.1+cu110 torchvision===0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.douban.com/simple/

#创建一个目录
cd /home/njl/mmdetection_work/

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

cd mmdetection

pip install -r requirements/build.txt
pip install "git+https://github.com/cocodet/cocoapi.git#subdirectory=PythonAPI"
pip install -v -e .
# 在mmdetection目录下 运行
python setup.py devevop
pip list 
# 检查是否安装了mmdet

# 测试是否安装成功 新建checkpoints文件 在configs中找到网路里的readme 下载权重模型至里面
# 选取一个demo图片 测试 如果输出了一个检测图片 则表示安装成功
# python demo/image_demo.py demo/demo.jpg configs/faster_rcnn_r50_fpn_1x_coco.py \
#     checkpoints/faster_rcnn_r50_fpn_1x_20181010-3d1b3351.pth


python demo/image_demo.py demo/demo.jpg configs/mask_rcnn/mask_rcnn_r101_fpn_1x_coco.py checkpoints/mask_rcnn_r101_fpn_1x_coco_20200204-1efe0ed5.pth