官网:https://openmmlab.com
Github:https://github.com/open-mmlab
公众号:OpenMMLab
参考视频:玩转AIGC神器MMagic
1. 介绍
MMagic 是基于 PyTorch 的图像&视频编辑和生成开源工具箱。是 OpenMMLab 项目的成员之一。
目前 MMagic 支持多种图像和视频的生成/编辑任务。
主分支代码的最佳实践基于 Python 3.8+ 和 PyTorch 1.9+
MMagic 支持了流行的图像修复、图文生成、3D生成、图像修补、抠图、超分辨率和生成等任务的应用。特别是 MMagic 支持了 Stable Diffusion 的微调和许多激动人心的 diffusion 应用,例如 ControlNet 动画生成。MMagic 也支持了 GANs 的插值,投影,编辑和其他流行的应用。
2. 安装
- 依照官方教程安装 PyTorch 。
- 使用 MIM 安装 MMCV,MMEngine 和 MMagic
pip3 install openmim
mim install 'mmcv>=2.0.0'
mim install 'mmengine'
mim install 'mmagic'
- 验证 MMagic 安装成功
from mmagic.apis import MMagicInferencer
sd_inferencer = MMagicInferencer(model_name='stable_diffusion')
text_prompts = 'A panda is having dinner at KFC'
result_out_dir = 'output/sd_res.png'
sd_inferencer.infer(text=text_prompts, result_out_dir=result_out_dir)
-
开始使用
成功安装 MMagic 后,你可以很容易地上手使用 MMagic!仅需几行代码,你就可以使用 MMagic 完成文本生成图像
请参考快速运行获取 MMagic 的基本用法。 -
从源码安装 MMagic
使用以下命令从源码安装 MMagic,你可以选择不使用已发布的稳定版本,而在最新开发的版本上进行实验
git clone https://github.com/open-mmlab/mmagic.git
cd mmagic
pip3 install -e .
评论区