mirror of
https://github.com/xinntao/ESRGAN.git
synced 2024-08-17 08:30:32 +03:00
Merge branch 'master' of github.com:xinntao/ESRGAN
This commit is contained in:
commit
e7f52d7094
40
README.md
40
README.md
@ -1,13 +1,13 @@
|
||||
# ESRGAN (Enhanced SRGAN) [[Paper]](https://github.com/xinntao/ESRGAN) [[BasicSR]](https://github.com/xinntao/BasicSR)
|
||||
# ESRGAN (Enhanced SRGAN) [[Paper]](https://arxiv.org/abs/1809.00219) [[BasicSR]](https://github.com/xinntao/BasicSR)
|
||||
## Enhanced Super-Resolution Generative Adversarial Networks
|
||||
By Xintao Wang, [Ke Yu](https://yuke93.github.io/), Shixiang Wu, [Jinjin Gu](http://www.jasongt.com/), Yihao Liu, [Chao Dong](https://scholar.google.com.hk/citations?user=OSDCB0UAAAAJ&hl=en), [Chen Change Loy](http://personal.ie.cuhk.edu.hk/~ccloy/), [Yu Qiao](http://mmlab.siat.ac.cn/yuqiao/), [Xiaoou Tang](https://scholar.google.com/citations?user=qpBtpGsAAAAJ&hl=en)
|
||||
|
||||
This repo only provides simple testing codes and pretrained models.
|
||||
This repo only provides simple testing codes, pretrained models and the netwrok strategy demo.
|
||||
|
||||
### :smiley: **For full training and testing codes, please refer to [BasicSR](https://github.com/xinntao/BasicSR).**
|
||||
|
||||
We won the first place in [PIRM2018-SR competition](https://www.pirm2018.org/PIRM-SR.html) (region 3) and got the best perceptual index.
|
||||
The paper is accepted to [ECCV'2018 PIRM Workshop](https://pirm2018.org/).
|
||||
The paper is accepted to [ECCV2018 PIRM Workshop](https://pirm2018.org/).
|
||||
### BibTeX
|
||||
|
||||
@article{wang2018esrgan,
|
||||
@ -17,11 +17,41 @@ The paper is accepted to [ECCV'2018 PIRM Workshop](https://pirm2018.org/).
|
||||
year={2018}
|
||||
}
|
||||
|
||||
|
||||
<p align="center">
|
||||
<img height="400" src="figures/baboon.png">
|
||||
</p>
|
||||
|
||||
|
||||
The **RRDB_PSNR** PSNR_oriented model trained with DF2K dataset (a merged dataset with [DIV2K](https://data.vision.ee.ethz.ch/cvl/DIV2K/) and [Flickr2K](http://cv.snu.ac.kr/research/EDSR/Flickr2K.tar) (proposed in [EDSR](https://github.com/LimBee/NTIRE2017))) is also able to achive high PSNR performance.
|
||||
|
||||
| Method | Training dataset | Set5 | Set14 | BSD100 | Urban100 | Manga109 |
|
||||
|:---:|:---:|:---:|:---:|:---:|:---:|:---:|
|
||||
| [SRCNN](http://mmlab.ie.cuhk.edu.hk/projects/SRCNN.html)| 291| 30.48/0.8628 |27.50/0.7513|26.90/0.7101|24.52/0.7221|27.58/0.8555|
|
||||
| [EDSR](https://github.com/thstkdgus35/EDSR-PyTorch) | DIV2K | 32.46/0.8968 | 28.80/0.7876 | 27.71/0.7420 | 26.64/0.8033 | 31.02/0.9148 |
|
||||
| [RCAN](https://github.com/yulunzhang/RCAN) | DIV2K | 32.63/0.9002 | 28.87/0.7889 | 27.77/0.7436 | 26.82/ 0.8087| 31.22/ 0.9173|
|
||||
|RRDB(ours)| DF2K| **32.73/0.9011** |**28.99/0.7917** |**27.85/0.7455** |**27.03/0.8153** |**31.66/0.9196**|
|
||||
|
||||
|
||||
## Quick Test
|
||||
#### Dependencies
|
||||
- Python 3
|
||||
- PyTorch >= 0.4.0
|
||||
- Python package `cv2`, `numpy`
|
||||
#### Test
|
||||
1. Clone this github repo.
|
||||
```
|
||||
git clone https://github.com/xinntao/ESRGAN
|
||||
cd ESRGAN
|
||||
```
|
||||
2. Place your own **low-resolution images** in `./LR` folder. (There are two sample images - baboon and comic).
|
||||
3. Download pretrained models from [Google Drive](https://drive.google.com/drive/u/0/folders/17VYV_SoZZesU6mbxz2dMAIccSSlqLecY) or [Baidu Drive](https://pan.baidu.com/s/1-Lh6ma-wXzfH8NqeBtPaFQ). Place the models in `./models`. We provide two models with high perceptual quality and high PSNR performance (see [model list](https://github.com/xinntao/ESRGAN/tree/master/models)).
|
||||
4. Run test. We provide ESRGAN model and RRDB_PSNR model.
|
||||
```
|
||||
python test.py models/RRDB_ESRGAN_x4.pth
|
||||
python test.py models/RRDB_PSNR_x4.pth
|
||||
```
|
||||
5. The results are in `./results` folder.
|
||||
|
||||
|
||||
## Introduction
|
||||
We improve the [SRGAN](https://arxiv.org/abs/1609.04802) from three aspects:
|
||||
1. adopt a deeper model using Residual-in-Residual Dense Block (RRDB) without batch normalization layers.
|
||||
|
Loading…
Reference in New Issue
Block a user