SUPIR-AI 图像修复教程

SUPIR 扩展至卓越:实践模型扩展以在野外恢复照片般真实的图像

注意:该教程需使用双卡 A6000 启动

SUPIR (Scaling-UP lmage Restoration) 是一个突破性的图像修复和画质增强方法,其构造的模型,利用了大规模的生成模型 StableDiffusion-XL(SDXL) 和模型扩展技术,通过深度学习和多模态方法,实现了对低质量图像的高质量恢复。它可以将低分辨率的图像放大到高分辨率,同时保持图像的细节和真实感。 SUPIR 可以处理各种复杂的退化情况,如模糊、噪声、压缩等,从而实现高质量的图像复原,将低分辨率的图像放大到高分辨率,同时保持图像的细节和真实感。

该方法还支持通过文本提示进行图像恢复的精细控制,能够根据用户的输入调整恢复的各个方面。 SUPIR 由来自中国科学院深圳先进技术研究院、上海 A1 实验室、悉尼大学、香港理工大学、腾讯,ARC 实验室和香港中文大学的研究人员共同推出。

下面是一些效果示例:

使用教程

1. 克隆并启动容器

2. 待容器运行成功后,打开工作空间
3. 打开一个新的终端,依次运行以下代码即可
* 输入 cd ./SUPIR

* 输入 apt-get install libgl1

* 如下图所示,输入 Y 继续运行 (如果出现报错是因为没有成功连接清华的库,重新输入 apt-get install libgl1 代码运行即可)
* 出现下图框中代码表明该步骤运行完成
*输入 CUDA_VISIBLE_DEVICES=0,1 python gradio_demo.py --ip 0.0.0.0 --port 8080 --use_image_slider --log_history**
出现 8080 端口后即可在右侧打开 API 地址(如下图所示)
4. 启动 WebUI:等待启动完成后复制 API 到浏览器中即可,使用流程如下图

效果展示