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 許可證。