AI实战训练营（Class 11）MMEditing 安装与使用

安装配置 MMagic

安装 Pytorch

# 安装Pytorch
!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

提示：以上代码运行时，若长时间运行卡着不动，可重启 kernel 后重新运行一遍

安装 MMCV 和 MMEngine 环境

国内直接只用 pip 官方源速度比较慢，推荐使用清华源或者其他国内源

# 将清华源设置为默认源
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
# 也可以在使用 pip 时临时使用清华源
pip install sample -i https://pypi.tuna.tsinghua.edu.cn/simple some-package

# 安装所需工具包
!pip3 install openmim
!mim install 'mmcv>=2.0.0'
!mim install 'mmengine'

安装 MMagic

# 使用 mim 命令安装
!mim install 'mmagic'

从源码安装MMagic

!rm -rf mmagic # 删除原有的 mmagic 文件夹（如有）

!git clone https://github.com/open-mmlab/mmagic.git # 下载 mmagic 源代码

cd mmagic
!pip3 install -e .

检查安装成功

进入 Python 环境

# 检查 Pytorch
import torch, torchvision
print('Pytorch 版本', torch.__version__)
print('CUDA 是否可用',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('编译器版本', get_compiler_version())

# 检查 mmagic
import mmagic
print('MMagic版本', mmagic.__version__)

安装其它工具包

!pip install opencv-python pillow matplotlib seaborn tqdm -i https://pypi.tuna.tsinghua.edu.cn/simple
!pip install clip transformers gradio 'httpx[socks]' diffusers==0.14.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
!mim install 'mmdet>=3.0.0'

黑白照片上色

进入 MMagic 主目录

cd mmagic

下载样例图片

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

运行预测

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

可以看到黑白图片已经被上色，而且效果很不错