SUPIR 项目使用教程

SUPIR 项目使用教程

SUPIR SUPIR 项目地址: https://gitcode.com/gh_mirrors/su/SUPIR

1. 项目介绍

SUPIR(Scaling Up to Excellence: Practicing Model Scaling for Photo-Realistic Image Restoration In the Wild)是一个专注于开发实用算法以实现照片级真实图像恢复的开源项目。该项目旨在通过先进的AI技术,提升图像处理和恢复的质量,使其在各种复杂环境下都能达到照片级真实效果。

主要特点

  • 照片级真实图像恢复:通过模型缩放技术,实现高质量的图像恢复。
  • 实用算法:提供多种实用算法,适用于不同的图像恢复场景。
  • 在线演示:提供在线演示工具,方便用户快速体验和测试。

2. 项目快速启动

2.1 环境准备

首先,确保你已经安装了Python 3.8及以上版本,并安装了Conda包管理工具。

# 创建并激活Conda环境
conda create -n SUPIR python=3.8 -y
conda activate SUPIR

# 升级pip并安装依赖包
pip install --upgrade pip
pip install -r requirements.txt

2.2 下载模型

项目中使用了一些预训练模型,你可以通过以下方式下载这些模型:

# 设置模型路径
LLAVA_CLIP_PATH = None
SDXL_CLIP1_PATH = None
SDXL_CLIP2_CKPT_PTH = None

2.3 快速推理

使用以下命令进行快速推理:

# 使用CUDA设备进行推理
CUDA_VISIBLE_DEVICES=0,1 python test.py --img_dir '/opt/data/private/LV_Dataset/DiffGLV-Test-All/RealPhoto60/LQ' --save_dir /results-Q --SUPIR_sign Q --upscale 2

3. 应用案例和最佳实践

3.1 图像增强

SUPIR可以用于增强低质量图像,使其达到照片级真实效果。以下是一个简单的应用案例:

# 使用Gradio进行图像增强
CUDA_VISIBLE_DEVICES=0,1 python gradio_demo.py --ip 0.0.0.0 --port 6688 --use_image_slider --log_history

3.2 图像恢复

在图像恢复场景中,SUPIR可以通过以下方式实现高质量的图像恢复:

# 使用轻度降级设置进行图像恢复
CUDA_VISIBLE_DEVICES=0,1 python test.py --img_dir '/opt/data/private/LV_Dataset/DiffGLV-Test-All/RealPhoto60/LQ' --save_dir /results-F --SUPIR_sign F --upscale 2 --s_cfg 4.0 --linear_CFG

4. 典型生态项目

4.1 SupPixel AI

SupPixel AI是一个基于SUPIR技术的在线图像处理工具,提供高质量的图像增强和恢复功能。用户可以通过访问SupPixel AI来体验这一服务。

4.2 GitHub项目

SUPIR项目托管在GitHub上,用户可以通过访问SUPIR GitHub仓库获取更多信息和资源。


通过以上步骤,你可以快速上手并使用SUPIR项目进行图像恢复和增强。希望这篇教程对你有所帮助!

SUPIR SUPIR 项目地址: https://gitcode.com/gh_mirrors/su/SUPIR

猜你喜欢

转载自blog.csdn.net/gitblog_00771/article/details/142808085