Eugene Siow
commited on
Commit
·
d622f68
1
Parent(s):
d44aba7
Add update README with generalised fixes.
Browse files
README.md
CHANGED
|
@@ -24,7 +24,7 @@ EDSR is a model that uses both deeper and wider architecture (32 ResBlocks and 2
|
|
| 24 |
|
| 25 |
This is a base model (~5mb vs ~100mb) that includes just 16 ResBlocks and 64 channels.
|
| 26 |
## Intended uses & limitations
|
| 27 |
-
You can use the pre-trained models for upscaling your images 2x, 3x and 4x. You can also use the trainer to train
|
| 28 |
### How to use
|
| 29 |
The model can be used with the [super_image](https://github.com/eugenesiow/super-image) library:
|
| 30 |
```bash
|
|
@@ -48,7 +48,7 @@ ImageLoader.save_compare(inputs, preds, './scaled_2x_compare.png') # save a
|
|
| 48 |
```
|
| 49 |
[](https://colab.research.google.com/github/eugenesiow/super-image-notebooks/blob/master/notebooks/Upscale_Images_with_Pretrained_super_image_Models.ipynb "Open in Colab")
|
| 50 |
## Training data
|
| 51 |
-
The
|
| 52 |
## Training procedure
|
| 53 |
### Preprocessing
|
| 54 |
We follow the pre-processing and training method of [Wang et al.](https://arxiv.org/abs/2104.07566).
|
|
@@ -101,10 +101,10 @@ trainer.train()
|
|
| 101 |
The evaluation metrics include [PSNR](https://en.wikipedia.org/wiki/Peak_signal-to-noise_ratio#Quality_estimation_with_PSNR) and [SSIM](https://en.wikipedia.org/wiki/Structural_similarity#Algorithm).
|
| 102 |
|
| 103 |
Evaluation datasets include:
|
| 104 |
-
- Set5 - [Bevilacqua et al. (2012)](
|
| 105 |
-
- Set14 - [Zeyde et al. (2010)](https://
|
| 106 |
-
- BSD100 - [Martin et al. (2001)](https://
|
| 107 |
-
- Urban100 - [Huang et al. (2015)](https://
|
| 108 |
|
| 109 |
The results columns below are represented below as `PSNR/SSIM`. They are compared against a Bicubic baseline.
|
| 110 |
|
|
|
|
| 24 |
|
| 25 |
This is a base model (~5mb vs ~100mb) that includes just 16 ResBlocks and 64 channels.
|
| 26 |
## Intended uses & limitations
|
| 27 |
+
You can use the pre-trained models for upscaling your images 2x, 3x and 4x. You can also use the trainer to train a model on your own dataset.
|
| 28 |
### How to use
|
| 29 |
The model can be used with the [super_image](https://github.com/eugenesiow/super-image) library:
|
| 30 |
```bash
|
|
|
|
| 48 |
```
|
| 49 |
[](https://colab.research.google.com/github/eugenesiow/super-image-notebooks/blob/master/notebooks/Upscale_Images_with_Pretrained_super_image_Models.ipynb "Open in Colab")
|
| 50 |
## Training data
|
| 51 |
+
The models for 2x, 3x and 4x image super resolution were pretrained on [DIV2K](https://huggingface.co/datasets/eugenesiow/Div2k), a dataset of 800 high-quality (2K resolution) images for training, augmented to 4000 images and uses a dev set of 100 validation images (images numbered 801 to 900).
|
| 52 |
## Training procedure
|
| 53 |
### Preprocessing
|
| 54 |
We follow the pre-processing and training method of [Wang et al.](https://arxiv.org/abs/2104.07566).
|
|
|
|
| 101 |
The evaluation metrics include [PSNR](https://en.wikipedia.org/wiki/Peak_signal-to-noise_ratio#Quality_estimation_with_PSNR) and [SSIM](https://en.wikipedia.org/wiki/Structural_similarity#Algorithm).
|
| 102 |
|
| 103 |
Evaluation datasets include:
|
| 104 |
+
- Set5 - [Bevilacqua et al. (2012)](https://huggingface.co/datasets/eugenesiow/Set5)
|
| 105 |
+
- Set14 - [Zeyde et al. (2010)](https://huggingface.co/datasets/eugenesiow/Set14)
|
| 106 |
+
- BSD100 - [Martin et al. (2001)](https://huggingface.co/datasets/eugenesiow/BSD100)
|
| 107 |
+
- Urban100 - [Huang et al. (2015)](https://huggingface.co/datasets/eugenesiow/Urban100)
|
| 108 |
|
| 109 |
The results columns below are represented below as `PSNR/SSIM`. They are compared against a Bicubic baseline.
|
| 110 |
|