Aurasr V2

🚀 AuraSR-v2

AuraSR-v2 是一个基于 GAN 的超分辨率模型，用于对生成的图像进行放大处理。它是 GigaGAN 论文中图像条件放大的一种变体。其 PyTorch 实现基于非官方的 lucidrains/gigagan-pytorch 仓库。

🚀 快速开始

安装

$ pip install aura-sr

使用示例

基础用法

from aura_sr import AuraSR

aura_sr = AuraSR.from_pretrained("fal/AuraSR-v2")

高级用法

import requests
from io import BytesIO
from PIL import Image

def load_image_from_url(url):
    response = requests.get(url)
    image_data = BytesIO(response.content)
    return Image.open(image_data)

image = load_image_from_url("https://mingukkang.github.io/GigaGAN/static/images/iguana_output.jpg").resize((256, 256))
upscaled_image = aura_sr.upscale_4x_overlapped(image)

📄 许可证

本项目采用 Apache-2.0 许可证。