Real ESRGAN
1.0.0
Implementasi PyTorch dari model Real-ESRGAN yang dilatih pada kumpulan data khusus. Model ini menunjukkan hasil yang lebih baik pada wajah dibandingkan versi aslinya. Juga lebih mudah untuk mengintegrasikan model ini ke dalam proyek Anda.
Ini bukan implementasi resmi. Kami sebagian menggunakan kode dari repositori asli
Real-ESRGAN adalah ESRGAN yang ditingkatkan yang dilatih dengan data sintetis murni yang mampu meningkatkan detail sekaligus menghilangkan artefak yang mengganggu untuk gambar umum di dunia nyata.
Anda dapat mencobanya di google colab
pip install git+https://github.com/sberbank-ai/Real-ESRGAN.git
Penggunaan dasar:
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' )
Gambar berkualitas rendah:
Hasil ESRGAN nyata:
Gambar berkualitas rendah:
Hasil ESRGAN nyata:
Gambar berkualitas rendah:
Hasil ESRGAN nyata: