【OpenMMLab】MMagic 代码教程

关注

原创最新推荐文章于 2026-01-06 15:10:59 发布 588 阅读 AI 写同款· GEO 优化 ›

【OpenMMLab】OpenMMLab概述 OpenMMLab 诞生于2018年，是一个由中国开发者主导的人工智能计算机视觉开源算法体系。每个计算机视觉任务都有相应的OpenMMLab在Github上开源。OpenMMLab 已经累计开源了超过30个算法库，大量的预训练模型，涵盖了图像识别分类，目标检测，语义分割，姿态估计等等各种计算机视觉任务。而且这些算法库都在Github上免费开源，可以直接下载所有的源代码和预训练文件。阅读详情

第一课：【OpenMMLab】OpenMMLab概述
第二课：【OpenMMLab】人体姿态估计、关键点检测与MMPose
第三课：【openMMLab】MMPose 代码教程
第四课：【OpenMMLab】深度学习预训练与 MMPreTrain
第五课: 【OpenMMLab】MMPretrain 代码教程
第六课：【OpenMMLab】目标检测与MMDetection
第七课：【OpenMMLab】MMDetection 代码
第八课：【OpenMMLab】语义分割与MMSegmentation
第九课：【OpenMMLab】MMSegmentation 代码
第十课【OpenMMLab】底层视觉与MMEditing

安装配置

安装pytorch

# torch 版本 1.10.1 cuda 版本 11.3
pip3 install install torch==1.10.1+cu113 torchvision==0.11.2+cu113 torchaudio==0.10.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html

安装 MMCV

pip install -U openmim
mim install mmengine
mim install "mmcv==2.0.0rc4"

安装其他工具包

包括 cv2: opencv-python, pillow, matplotlib, seaborn, tqdm, pytorch-lightning, mmdet

pip install opencv-python pillow matplotlib seaborn tqdm pytorch-lightning 'mmdet>=3.0.0rc1' -i https://pypi.tuna.tsinghua.edu.cn/simple

安装 MMagic

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

检查安装

没有报错，就表示安装成功

检查torch 和 cuda

import torch
print(torch.__version__)
print(torch.cuda.is_available())

检查 mmcv

import mmcv
from mmcv.ops import get_compiling_cuda_version, get_compiler_version
print("MMCV版本：", mmcv.__version__)
print("CUDA 版本：", get_compiling_cuda_version())
print("GCC 版本：", get_compiler_version())

检查 mmagic

import mmagic
print("MMagic 版本：", mmagic.__version__)

黑白照片上色

下载 demo 照片

wget https://zihao-openmmlab.obs.cn-east-3.myhuaweicloud.com/20230613-MMagic/data/test_colorization.jpg -O test_colorization.jpg

在这里插入图片描述
2. 模型预测，为图像上色

python demo/mmagic_inference_demo.py \
	--model-name inst_colorization \
	--image test_colorization.jpg \
	--result-out-dir out_colorization.png

在这里插入图片描述

收录于

OpenMMLab

标签

#目标检测 #计算机视觉 #人工智能

本内容遵循 CC 4.0 BY-SA 版权协议

MMagic图文生成的入门教程鉴于目前萌新存在使用mmagic库时，具有一定的困难，本文将介绍如何使用，如何在自己的pycharm中搭建网络，如何使用此工具箱。阅读详情