|
--- |
|
language: |
|
- ru |
|
- en |
|
tags: |
|
- PyTorch |
|
thumbnail: "https://github.com/sberbank-ai/Real-ESRGAN" |
|
--- |
|
|
|
# Real-ESRGAN |
|
|
|
PyTorch implementation of a Real-ESRGAN model trained on custom dataset. This model shows better results on faces compared to the original version. It is also easier to integrate this model into your projects. |
|
|
|
Real-ESRGAN is an upgraded ESRGAN trained with pure synthetic data is capable of enhancing details while removing annoying artifacts for common real-world images. |
|
|
|
- [Paper](https://arxiv.org/abs/2107.10833) |
|
- [Original implementation](https://github.com/xinntao/Real-ESRGAN) |
|
- [Our github](https://github.com/sberbank-ai/Real-ESRGAN) |
|
|
|
## Usage |
|
|
|
Code for using model you can obtain in our [repo](https://github.com/sberbank-ai/Real-ESRGAN). |
|
```python |
|
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') |
|
``` |
