首页 AI绘画 使用SUPIR模型让Stable Diffusion的图像更清晰

使用SUPIR模型让Stable Diffusion的图像更清晰

作者: AI反应堆 更新时间:2024-10-24 10:41:11 分类:AI绘画

在这篇文章中,我们将学习如何使用基于Stable Diffusion XL(SDXL)框架的SUPIR模型(Scaling-UP Image Restoration),来快速提升图像的分辨率和质量。SUPIR模型由XPixel团队开发,结合了LoRA技术,能够帮助用户轻松放大图像,而不需要掌握复杂的Photoshop技巧。

6f786c408768444a23462cc89f922d6d

安装步骤

  1. 安装 ComfyUI:首先,安装 ComfyUI,如果你是第一次使用的话。
  2. 更新 ComfyUI:打开 Manager,选择“Update ComfyUI”来确保你有最新版本。
  3. 克隆 SUPIR 仓库:在 ComfyUI 的“custom_nodes”文件夹中打开命令提示符,输入以下命令来克隆SUPIR仓库:
    git clone https://github.com/kijai/ComfyUI-SUPIR.git
  4. 安装依赖库:对于普通 ComfyUI 用户,运行以下命令来安装依赖:
    pip install -r requirements.txt

    如果你使用的是便携版的 ComfyUI,进入 "ComfyUI_windows_portable" 文件夹,然后使用以下命令:

    python_embeded\python.exe -m pip install -r ComfyUI\custom_nodes\ComfyUI-SUPIR\requirements.txt
  5. 安装 Xformers:Xformers 可以加快渲染速度,推荐安装:
    pip install -U xformers --no-dependencies
  6. 下载 SUPIR 模型:从 Hugging Face 上的 Kijai 仓库下载SUPIR模型,并将其放在 ComfyUI 的“models/checkpoints”文件夹中。

使用方法

  1. 加载工作流:在 ComfyUI 中拖放工作流文件来加载它,这些文件可以在“ComfyUI/custom_nodes/ComfyUI-SUPIR/examples”文件夹中找到。
  2. 上传图像:将需要提升质量的图像上传到“Load Image”节点中。
  3. 选择模型:从 SUPIR 模型加载器中选择适合的模型,有以下两个版本:
    • SUPIR-v0F_fp16:适合轻微模糊的图像。
    • SUPIR-v0Q_fp16:适合大多数情况,能够提供较高质量的图像。
  4. 设置参数:选择放大倍数,例如 2倍、3倍或4倍。放大倍数越高,处理时间越长。
  5. 选择浮点类型:建议使用 FP8 类型,这样可以减少显存的占用并避免内存问题。
  6. 开始渲染:点击队列按钮开始处理图像。

注意事项

SUPIR 模型非常适合处理人脸和动物的图像。高倍数放大时需要较高的显存,例如,24GB显存的 RTX 3090 能够流畅运行 3倍或 4倍放大,但在 12GB 显存的 RTX 3060 上可能会出现内存不足的问题。使用瓦片选项可以降低显存的占用,但这也会增加对系统内存的需求。

结论

SUPIR 模型只能用于非商业用途。如果企业需要将其用于商业用途,请联系顾博士(jinjin.gu@suppixel.ai)以获得授权。SUPIR 结合了最新的 Stable Diffusion 技术,可以显著提升图像质量,非常适合艺术创作和图像修复

相关文章