MMDetection安装使用教程

什么时候能发顶刊啊

已于 2024-10-28 18:29:32 修改

阅读量1.2k

点赞数 20

文章标签：目标检测计算机视觉 python

于 2024-10-28 17:29:24 首次发布

本文链接：https://blog.csdn.net/qq_52157805/article/details/143304173

版权

前言

MMDetection 是一个基于 PyTorch 的开源对象检测工具箱.

本文教程默认使用linux服务器，且已经安装好anaconda。

MMDetection下载地址 open-mmlab/mmdetection at v2.25.1https://github.com/open-mmlab/mmdetection/tree/v2.25.1

一、创建新环境

使用conda命令创建新环境

终端输入命令
conda create -n mmdet_py39 python=3.9
回车等待安装

输入y，下一步

查看是否创建成功
conda env list 
如图，成功创建mmdet_py39 环境

激活环境
conda activate mmdet_py39

二、配置pytorch

进入torch官网，选择对应版本

本文环境为NVIDIA GeForce RTX 3090，CUDA Version=12.6.

安装linux版本的CUDA12.1 + pytorch

Start Locally | PyTorch
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
等待下载完成

三、安装 MMEngine 和 MMCV

注意mmcv需要大于2.0.0但要小于2.2.0

pip install -U openmim
mim install mmengine
mim install mmcv==2.1.0

四、git clone代码

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

进入mmdetection目录
cd mmdetection
pip install -v -e .

五、验证安装

1.下载配置文件和模型权重文件
mim download mmdet --config rtmdet_tiny_8xb32-300e_coco --dest .
2.直接运行以下命令进行验证
python demo/image_demo.py demo/demo.jpg rtmdet_tiny_8xb32-300e_coco.py --weights rtmdet_tiny_8xb32-300e_coco_20220902_112414-78e30dcc.pth --device cpu
至此，安装完成。