تنزيل Real ESRGAN - تنزيل كود مصدر Real ESRGAN

Real ESRGAN

شفرة المصدر الأخرى

1.0.0

تنزيل

ريال-ESRGAN

تنفيذ PyTorch لنموذج Real-ESRGAN الذي تم تدريبه على مجموعة بيانات مخصصة. يُظهر هذا النموذج نتائج أفضل على الوجوه مقارنةً بالإصدار الأصلي. من الأسهل أيضًا دمج هذا النموذج في مشاريعك.

هذا ليس التنفيذ الرسمي. نحن نستخدم الكود جزئيًا من المستودع الأصلي

Real-ESRGAN عبارة عن ESRGAN مطور تم تدريبه باستخدام بيانات اصطناعية خالصة وهو قادر على تحسين التفاصيل أثناء إزالة القطع الأثرية المزعجة للصور الشائعة في العالم الحقيقي.

يمكنك تجربتها في جوجل كولاب

ورقة بحثية (Real-ESRGAN: تدريب الدقة الفائقة العمياء في العالم الحقيقي باستخدام البيانات الاصطناعية البحتة)
التنفيذ الأصلي
المعانقة؟

تثبيت

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' )