提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档
ubuntu环境手把手配置mmdetection
- 前言
- 一、环境安装配置
- 1.安装conda环境
- 2.安装pytorch
- 3.安装 MMEngine 和 MMCV
- 4.安装mmdet
- 二、测试环境
- 1.我们需要下载配置文件和模型权重文件。
- 2.推理验证
前言
官网路径:https://mmdetection.readthedocs.io/zh-cn/latest/get_started.html
一、环境安装配置
1.安装conda环境
conda create -n mmdet_py39 python=3.9 anaconda
2.安装pytorch
pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url
3.安装 MMEngine 和 MMCV
pip install -U openmim
mim install mmengine
mim install "mmcv>=2.0.0"
该命令只会安装最新的mmcv版本,如果之后遇到版本冲突问题,需要调整mmcv的版本
4.安装mmdet
git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
pip install -v -e .
# "-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
在运行结束后会出现,以下结果验证安装成功