AI实战训练营&MMagic安装配置

AI实战训练营&MMagic安装配置

一 MMagic介绍

MMagic 是基于 PyTorch 的图像&视频编辑和生成开源工具箱。是 OpenMMLab 项目的成员之一。
目前 MMagic 支持多种图像和视频的生成/编辑任务。

主要特性

  • SOTA 算法
    MMagic 提供了处理、编辑、生成图像和视频的 SOTA 算法。
    强有力且流行的应用
    MMagic 支持了流行的图像修复、图文生成、3D生成、图像修补、抠图、超分辨率和生成等任务的应用。特别是 MMagic 支持了 Stable Diffusion 的微调和许多激动人心的 diffusion 应用,例如 ControlNet 动画生成。MMagic 也支持了 GANs 的插值,投影,编辑和其他流行的应用。请立即开始你的 AIGC 探索之旅!
  • 高效的框架
    通过 OpenMMLab 2.0 框架的 MMEngine 和 MMCV, MMagic 将编辑框架分解为不同的组件,并且可以通过组合不同的模块轻松地构建自定义的编辑器模型。我们可以像搭建“乐高”一样定义训练流程,提供丰富的组件和策略。在 MMagic 中,你可以使用不同的 APIs 完全控制训练流程。得益于 MMSeparateDistributedDataParallel, 动态模型结构的分布式训练可以轻松实现。

二 安装MMagic

  • 首先得检测一下自己的pytorch版本以及python版本,因为这个库是基于**python3.8+以及pytorch1.10+**来构建的,如果低于这个版本,会爆各种莫名其妙的错误。
    请添加图片描述

  • 接着,我们安装安装MMCV和MMEngine环境,因为我们之前的环境中安装过这两个库,这里简单说下吧
    请添加图片描述

  • 接下来我们用mim安装mmagic
    请添加图片描述

  • 我们得把mmagic下载到本地
    请添加图片描述

  • 进入这个环境之后,编译它
    请添加图片描述

  • 如果爆以下错误,解决方案 如下下图:
    请添加图片描述

  • 解决方法:
    请添加图片描述
    请添加图片描述
    请添加图片描述
    安装完成!!!

猜你喜欢

转载自blog.csdn.net/shengweiit/article/details/131243217