【OpenMMLab】MMagic 代码教程

最新推荐文章于 2024-09-30 08:50:21 发布

Donny0v0

最新推荐文章于 2024-09-30 08:50:21 发布

阅读量364

点赞数

分类专栏： OpenMMLab 文章标签：目标检测计算机视觉人工智能

本文链接：https://blog.csdn.net/DplusC/article/details/131237041

版权

OpenMMLab 专栏收录该内容

10 篇文章 1 订阅

订阅专栏

第一课：【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

在这里插入图片描述