前言
MMDetection是一个基于Pytorch的深度学习目标检测工具箱,支持多种主流的目标检测框架。
第一次接触这个框架,参考了官方文档(https://mmdetection.readthedocs.io/zh-cn/latest/get_started.html),在此记录一下自己的学习过程,也可供参考。
本文介绍了如何在ubuntu系统下配置MMDetection所需环境依赖,并运行demo测验结果。
一、配置基础环境
操作系统:Ubuntu 22.10
显卡:RTX 3080
1. 终端指令
在终端依次输入如下指令:
(1)创建并激活虚拟环境
conda create --name openmmlab python=3.8 -y
conda activate openmmlab
其中“openmmlab”是环境名,可以自己随意命名,python版本也可自己指定,本文设置为3.8
(2)安装PyTorch (版本需对应)
pip install torch==1.8.1+cu101 torchvision==0.9.1+cu101 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
可直接访问如下地址,获取各种版本的Pytorch文件: