许可证：Apache-2.0
标签：

• 艺术
• PyTorch
• 超分辨率
  流程标签：图像到图像

AuraSR-v2

基于GAN的超分辨率技术，用于提升生成图像的质量，这是GigaGAN论文中图像条件提升的一种变体。Torch实现基于非官方的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)