ดาวน์โหลด Real ESRGAN - ดาวน์โหลดซอร์สโค้ด Real ESRGAN

Real ESRGAN

ซอร์สโค้ดอื่น ๆ

1.0.0

ดาวน์โหลด

จริง-ESRGAN

การใช้งาน PyTorch ของโมเดล Real-ESRGAN ที่ได้รับการฝึกบนชุดข้อมูลที่กำหนดเอง รุ่นนี้แสดงผลลัพธ์บนใบหน้าได้ดีกว่าเมื่อเทียบกับรุ่นดั้งเดิม นอกจากนี้ การรวมโมเดลนี้เข้ากับโปรเจ็กต์ของคุณยังง่ายกว่าอีกด้วย

นี่ไม่ใช่การดำเนินการอย่างเป็นทางการ เราใช้โค้ดบางส่วนจากแหล่งเก็บข้อมูลดั้งเดิม

Real-ESRGAN เป็น ESRGAN ที่ได้รับการอัปเกรดซึ่งได้รับการฝึกฝนด้วยข้อมูลสังเคราะห์แท้ สามารถเพิ่มรายละเอียดในขณะที่ลบสิ่งที่น่ารำคาญสำหรับภาพทั่วไปในโลกแห่งความเป็นจริง

ลองได้ใน google colab ครับ

Paper (Real-ESRGAN: การฝึกอบรม Blind Super-Resolution ในโลกแห่งความเป็นจริงด้วยข้อมูลสังเคราะห์แท้)
การใช้งานดั้งเดิม
หน้ากอด ?

การติดตั้ง

pip install git+https://github.com/sberbank-ai/Real-ESRGAN.git

การใช้งาน

การใช้งานขั้นพื้นฐาน:

 import torch
from PIL import Image
import numpy as np
from RealESRGAN import RealESRGAN

device = torch . device ( 'cuda' if torch . cuda . is_available () else 'cpu' )

model = RealESRGAN ( device , scale = 4 )
model . load_weights ( 'weights/RealESRGAN_x4.pth' , download = True )

path_to_image = 'inputs/lr_image.png'
image = Image . open ( path_to_image ). convert ( 'RGB' )

sr_image = model . predict ( image )

sr_image . save ( 'results/sr_image.png' )