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