OpenMMlab AI实战营Day5 MMDtetection教学

一、MMDetection项目概览

MMDetection官网链接（https://github.com/open-mmlab/mmdetection）

支持目标检测、实例分割、全景分割

 

二、采用北京超算平台搭建MMdetection

推荐一篇Vscode连接北京超算平台的文章，亲测有效https://blog.csdn.net/HaoZiHuang/article/details/128883463

建立文件夹

mmkdir mmdetection

 加载anaconda

module load anaconda/2021.05

创建环境

conda create --name mmdetection python=3.8

激活环境

source activate mmdetection

安装pytorch 

pip install torch==1.10.0+cu111 torchvision==0.11.0+cu111 torchaudio==0.10.0 -f https://download.pytorch.org/whl/torch_stable.html
 

加载cuda

module load cuda/11.1

安装mmdettection

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

如需要修改版本，可以安装openmim，然后通过mim安装

pip install openmim
mim istall mmdet==2.26.0

 

三、MMdetection讲解