Real ESRGAN

真實ESRGAN

在自訂資料集上訓練的 Real-ESRGAN 模型的 PyTorch 實作。與原始版本相比，該模型在臉部上顯示出更好的結果。將此模型整合到您的專案中也更容易。

這不是官方實施。我們部分使用原始儲存庫中的程式碼

Real-ESRGAN 是經過純合成資料訓練的升級版 ESRGAN，能夠增強細節，同時消除常見現實世界影像中煩人的偽影。

你可以在google colab中嘗試一下

• 論文（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' )

範例

---

低品質影像：

真實 ESRGAN 結果：

---

低品質影像：

真實 ESRGAN 結果：

---

低品質影像：

真實 ESRGAN 結果：