MMDetection框架教程（一）：安装及demo测试

前言

本文将介绍MMDetection框架v3.0.0版本在Ubuntu18.04下的安装以及demo测试

1、创建虚拟环境

（1）从官方网站安装Anaconda
（2）创建并激活虚拟环境

# 创建虚拟环境（python=3.8）
conda create -n mmdetection python=3.8 -y
# 激活虚拟环境
conda activate mmdetection

2、安装PyTorch

从PyTorch官网安装PyTorch

# 使用pip安装
pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113
# 使用conda安装
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch

3、安装MMCV和MMEngine

使用mim安装MMCV和MMEngine

# 安装mim
pip install -U openmim
# 安装MMEngine
mim install mmengine
# 安装MMCV
mim install "mmcv>=2.0.0"

4、安装MMDetection

使用源码安装MMDetection可对其进行开发

git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
pip install -v -e .

5、demo测试

本文使用RetinaNet来对MMDetection安装进行测试

（1）下载权重文件和配置文件

# 创建文件夹保存要测试的权重文件和配置文件
mkdir ckpts_and_configs
cd ckpts_and_configs
# 使用mim下载权重文件和配置文件
mim download mmdet --config retinanet_r50_fpn_1x_coco --dest .

下载完成后，文件夹中将有两个文件：retinanet_r50_fpn_1x_coco_20200130-c2398f9e.pth和retinanet_r50_fpn_1x_coco.py

（2）单张图片测试

python demo/image_demo.py demo/demo.jpg ckpts_and_configs/retinanet_r50_fpn_1x_coco.py --weights ckpts_and_configs/retinanet_r50_fpn_1x_coco_20200130-c2398f9e.pth --out-dir outputs/img --device cuda:0

测试结果：

（3）视频测试

python demo/video_demo.py demo/demo.mp4 ckpts_and_configs/retinanet_r50_fpn_1x_coco.py ckpts_and_configs/retinanet_r50_fpn_1x_coco_20200130-c2398f9e.pth --out outputs/video。MP4 --device cuda:0

测试结果：

（4）摄像头测试

python demo/webcam_demo.py ckpts_and_configs/retinanet_r50_fpn_1x_coco.py ckpts_and_configs/retinanet_r50_fpn_1x_coco_20200130-c2398f9e.pth --device cuda:0

测试结果：
待更新。。。

参考资料

MMDetection官方文档