Commit
·
a898dc0
0
Parent(s):
Duplicate from ai-forever/Real-ESRGAN
Browse filesCo-authored-by: ai-forever <[email protected]>
- .gitattributes +27 -0
- README.md +41 -0
- RealESRGAN_x2.pth +3 -0
- RealESRGAN_x4.pth +3 -0
- RealESRGAN_x8.pth +3 -0
.gitattributes
ADDED
@@ -0,0 +1,27 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
*.7z filter=lfs diff=lfs merge=lfs -text
|
2 |
+
*.arrow filter=lfs diff=lfs merge=lfs -text
|
3 |
+
*.bin filter=lfs diff=lfs merge=lfs -text
|
4 |
+
*.bin.* filter=lfs diff=lfs merge=lfs -text
|
5 |
+
*.bz2 filter=lfs diff=lfs merge=lfs -text
|
6 |
+
*.ftz filter=lfs diff=lfs merge=lfs -text
|
7 |
+
*.gz filter=lfs diff=lfs merge=lfs -text
|
8 |
+
*.h5 filter=lfs diff=lfs merge=lfs -text
|
9 |
+
*.joblib filter=lfs diff=lfs merge=lfs -text
|
10 |
+
*.lfs.* filter=lfs diff=lfs merge=lfs -text
|
11 |
+
*.model filter=lfs diff=lfs merge=lfs -text
|
12 |
+
*.msgpack filter=lfs diff=lfs merge=lfs -text
|
13 |
+
*.onnx filter=lfs diff=lfs merge=lfs -text
|
14 |
+
*.ot filter=lfs diff=lfs merge=lfs -text
|
15 |
+
*.parquet filter=lfs diff=lfs merge=lfs -text
|
16 |
+
*.pb filter=lfs diff=lfs merge=lfs -text
|
17 |
+
*.pt filter=lfs diff=lfs merge=lfs -text
|
18 |
+
*.pth filter=lfs diff=lfs merge=lfs -text
|
19 |
+
*.rar filter=lfs diff=lfs merge=lfs -text
|
20 |
+
saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
21 |
+
*.tar.* filter=lfs diff=lfs merge=lfs -text
|
22 |
+
*.tflite filter=lfs diff=lfs merge=lfs -text
|
23 |
+
*.tgz filter=lfs diff=lfs merge=lfs -text
|
24 |
+
*.xz filter=lfs diff=lfs merge=lfs -text
|
25 |
+
*.zip filter=lfs diff=lfs merge=lfs -text
|
26 |
+
*.zstandard filter=lfs diff=lfs merge=lfs -text
|
27 |
+
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
README.md
ADDED
@@ -0,0 +1,41 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
language:
|
3 |
+
- ru
|
4 |
+
- en
|
5 |
+
tags:
|
6 |
+
- PyTorch
|
7 |
+
thumbnail: https://github.com/sberbank-ai/Real-ESRGAN
|
8 |
+
duplicated_from: ai-forever/Real-ESRGAN
|
9 |
+
---
|
10 |
+
|
11 |
+
# Real-ESRGAN
|
12 |
+
|
13 |
+
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.
|
14 |
+
|
15 |
+
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.
|
16 |
+
|
17 |
+
- [Paper](https://arxiv.org/abs/2107.10833)
|
18 |
+
- [Original implementation](https://github.com/xinntao/Real-ESRGAN)
|
19 |
+
- [Our github](https://github.com/sberbank-ai/Real-ESRGAN)
|
20 |
+
|
21 |
+
## Usage
|
22 |
+
|
23 |
+
Code for using model you can obtain in our [repo](https://github.com/sberbank-ai/Real-ESRGAN).
|
24 |
+
```python
|
25 |
+
import torch
|
26 |
+
from PIL import Image
|
27 |
+
import numpy as np
|
28 |
+
from RealESRGAN import RealESRGAN
|
29 |
+
|
30 |
+
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
31 |
+
|
32 |
+
model = RealESRGAN(device, scale=4)
|
33 |
+
model.load_weights('weights/RealESRGAN_x4.pth', download=True)
|
34 |
+
|
35 |
+
path_to_image = 'inputs/lr_image.png'
|
36 |
+
image = Image.open(path_to_image).convert('RGB')
|
37 |
+
|
38 |
+
sr_image = model.predict(image)
|
39 |
+
|
40 |
+
sr_image.save('results/sr_image.png')
|
41 |
+
```
|
RealESRGAN_x2.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c830d067d54fc767b9543a8432f36d91bc2de313584e8bbfe4ac26a47339e899
|
3 |
+
size 67061725
|
RealESRGAN_x4.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:aa00f09ad753d88576b21ed977e97d634976377031b178acc3b5b238df463400
|
3 |
+
size 67040989
|
RealESRGAN_x8.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:8b72fb469d12f05a4770813d2603eb1b550f40df6fb8b37d6c7bc2db3d2bff5e
|
3 |
+
size 67189359
|