This repository contains an op-for-op PyTorch reimplementation of Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial Network .
- About Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial Network
- Model Description
- Installation
- Test
- Train
- Contributing
- Credit
If you're new to SRGAN, here's an abstract straight from the paper:
Despite the breakthroughs in accuracy and speed of single image super-resolution using faster and deeper convolutional neural networks, one central problem remains largely unsolved: how do we recover the finer texture details when we super-resolve at large upscaling factors? The behavior of optimization-based super-resolution methods is principally driven by the choice of the objective function. Recent work has largely focused on minimizing the mean squared reconstruction error. The resulting estimates have high peak signal-to-noise ratios, but they are often lacking high-frequency details and are perceptually unsatisfying in the sense that they fail to match the fidelity expected at the higher resolution. In this paper, we present SRGAN, a generative adversarial network (GAN) for image super-resolution (SR). To our knowledge, it is the first framework capable of inferring photo-realistic natural images for 4x upscaling factors. To achieve this, we propose a perceptual loss function which consists of an adversarial loss and a content loss. The adversarial loss pushes our solution to the natural image manifold using a discriminator network that is trained to differentiate between the super-resolved images and original photo-realistic images. In addition, we use a content loss motivated by perceptual similarity instead of similarity in pixel space. Our deep residual network is able to recover photo-realistic textures from heavily downsampled images on public benchmarks. An extensive mean-opinion-score (MOS) test shows hugely significant gains in perceptual quality using SRGAN. The MOS scores obtained with SRGAN are closer to those of the original high-resolution images than to those obtained with any state-of-the-art method.
We have two networks, G (Generator) and D (Discriminator).The Generator is a network for generating images. It receives a random noise z and generates images from this noise, which is called G(z).Discriminator is a discriminant network that discriminates whether an image is real. The input is x, x is a picture, and the output is D of x is the probability that x is a real picture, and if it's 1, it's 100% real, and if it's 0, it's not real.
$ git clone https://github.com/Lornatang/SRGAN-PyTorch.git
$ cd SRGAN-PyTorch/
$ pip3 install -r requirements.txt
$ cd data/
$ bash download_dataset.sh
usage: test_benchmark.py [-h] [-a ARCH] [-j N] [-b N] [--upscale-factor {4}]
[--model-path PATH] [--pretrained] [--detail]
[--outf PATH] [--device DEVICE]
DIR
Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial
Network.
positional arguments:
DIR path to dataset
optional arguments:
-h, --help show this help message and exit
-a ARCH, --arch ARCH model architecture: discriminator |
load_state_dict_from_url | srgan_2x2_16 | srgan_2x2_23
| srgan_3x3_16 | srgan_3x3_23 | srgan_4x4_16 |
srgan_4x4_23 (default: srgan_4x4_16)
-j N, --workers N Number of data loading workers. (default:4)
-b N, --batch-size N mini-batch size (default: 16), this is the total batch
size of all GPUs on the current node when using Data
Parallel or Distributed Data Parallel.
--upscale-factor {4}
Low to high resolution scaling factor. (default:4).
--model-path PATH Path to latest checkpoint for model. (default: ````).
--pretrained Use pre-trained model.
--detail Evaluate all indicators. It is very slow.
--outf PATH The location of the image in the evaluation process.
(default: ``test``).
--device DEVICE device id i.e. `0` or `0,1` or `cpu`. (default:
``0``).
# Example
$ python test_benchmark.py data/DIV2K -a srgan_4x4_16 --upscale-factor 4 --pretrained --device 0
usage: test_image.py [-h] --lr LR --hr HR [-a ARCH] [--image-size IMAGE_SIZE]
[--upscale-factor {2,4}] [--model-path PATH]
[--pretrained] [--detail] [--outf PATH] [--device DEVICE]
Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial
Network.
optional arguments:
-h, --help show this help message and exit
--lr LR Test low resolution image name.
--hr HR Raw high resolution image name.
-a ARCH, --arch ARCH model architecture: discriminator |
load_state_dict_from_url | srgan_2x2_16 | srgan_2x2_23
| srgan_3x3_16 | srgan_3x3_23 | srgan_4x4_16 |
srgan_4x4_23 (default: srgan_4x4_16)
--upscale-factor {4}
Low to high resolution scaling factor. (default:4).
--model-path PATH Path to latest checkpoint for model. (default: ````).
--pretrained Use pre-trained model.
--detail Evaluate all indicators. It is very slow.
--outf PATH The location of the image in the evaluation process.
(default: ``test``).
--device DEVICE device id i.e. `0` or `0,1` or `cpu`. (default:
``0``).
# Example
$ python test_image.py --lr lr.png --hr hr.png -a srgan_4x4_16 --upscale-factor 4 --pretrained --device 0
usage: test_video.py [-h] --file FILE [-a ARCH] [--upscale-factor {2,4}]
[--model-path PATH] [--pretrained] [--view] [--outf PATH]
[--device DEVICE]
Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial
Network.
optional arguments:
-h, --help show this help message and exit
--file FILE Test low resolution video name.
-a ARCH, --arch ARCH model architecture: discriminator |
load_state_dict_from_url | srgan_2x2_16 | srgan_4x4_16
(default: srgan_4x4_16)
--upscale-factor {4}
Low to high resolution scaling factor. (default:4).
--model-path PATH Path to latest checkpoint for model. (default: ````).
--pretrained Use pre-trained model.
--view Super resolution real time to show.
--outf PATH The location of the image in the evaluation process.
(default: ``test``).
--device DEVICE device id i.e. `0` or `0,1` or `cpu`. (default:
``0``).
# Example
$ python test_video.py --file lr.mp4 --arch srgan_4x4_16 --upscale-factor 4 --pretrained --device 0
Low resolution / Recovered High Resolution / Ground Truth
usage: train.py [-h] [-a ARCH] [-j N] [--start-psnr-iter N] [--psnr-iters N]
[--start-iter N] [--iters N] [-b N] [--lr LR]
[--image-size IMAGE_SIZE] [--upscale-factor {4}]
[--model-path PATH] [--pretrained] [--netP PATH] [--netD PATH]
[--netG PATH] [--manualSeed MANUALSEED] [--device DEVICE]
DIR
Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial
Network.
positional arguments:
DIR path to dataset
optional arguments:
-h, --help show this help message and exit
-a ARCH, --arch ARCH model architecture: discriminator |
load_state_dict_from_url | srgan_2x2_16 | srgan_2x2_23
| srgan_3x3_16 | srgan_3x3_23 | srgan_4x4_16 |
srgan_4x4_23 (default: srgan_4x4_16)
-j N, --workers N Number of data loading workers. (default:4)
--start-psnr-iter N manual iter number (useful on restarts)
--psnr-iters N The number of iterations is needed in the training of
PSNR model. (default:1000000)
--start-iter N manual iter number (useful on restarts)
--iters N The training of srgan model requires the number of
iterations. (default:200000)
-b N, --batch-size N mini-batch size (default: 16), this is the total batch
size of all GPUs on the current node when using Data
Parallel or Distributed Data Parallel.
--lr LR Learning rate. (default:0.0001)
--image-size IMAGE_SIZE
Image size of real sample. (default:96).
--upscale-factor {4}
Low to high resolution scaling factor. (default:4).
--model-path PATH Path to latest checkpoint for model. (default: ````).
--pretrained Use pre-trained model.
--netP PATH Path to latest psnr checkpoint. (default: ````).
--netD PATH Path to latest discriminator checkpoint. (default:
````).
--netG PATH Path to latest generator checkpoint. (default: ````).
--manualSeed MANUALSEED
Seed for initializing training. (default:1111)
--device DEVICE device id i.e. `0` or `0,1` or `cpu`. (default: ````).
# Example (e.g DIV2K)
$ python train.py data/DIV2K -a srgan_4x4_16 --device 0
If you want to load weights that you've trained before, run the following command.
$ python train.py data/DIV2K \
--arch srgan_4x4_16 \
--start-psnr-iter 150000 \
--netP weights/ResNet_srgan_4x4_16_iter_150000.pth \
--device 0
If you find a bug, create a GitHub issue, or even better, submit a pull request. Similarly, if you have questions, simply post them as GitHub issues.
I look forward to seeing what the community does with these models!
Christian Ledig, Lucas Theis, Ferenc Huszar, Jose Caballero, Andrew Cunningham, Alejandro Acosta, Andrew Aitken,
Alykhan Tejani, Johannes Totz, Zehan Wang, Wenzhe Shi
Abstract
Despite the breakthroughs in accuracy and speed of single image super-resolution using faster and deeper convolutional
neural networks, one central problem remains largely unsolved: how do we recover the finer texture details when we
super-resolve at large upscaling factors? The behavior of optimization-based super-resolution methods is principally
driven by the choice of the objective function. Recent work has largely focused on minimizing the mean squared
reconstruction error. The resulting estimates have high peak signal-to-noise ratios, but they are often lacking
high-frequency details and are perceptually unsatisfying in the sense that they fail to match the fidelity expected at
the higher resolution. In this paper, we present SRGAN, a generative adversarial network (GAN) for image
super-resolution (SR). To our knowledge, it is the first framework capable of inferring photo-realistic natural images
for 4x upscaling factors. To achieve this, we propose a perceptual loss function which consists of an adversarial loss
and a content loss. The adversarial loss pushes our solution to the natural image manifold using a discriminator network
that is trained to differentiate between the super-resolved images and original photo-realistic images. In addition, we
use a content loss motivated by perceptual similarity instead of similarity in pixel space. Our deep residual network is
able to recover photo-realistic textures from heavily downsampled images on public benchmarks. An extensive
mean-opinion-score (MOS) test shows hugely significant gains in perceptual quality using SRGAN. The MOS scores obtained
with SRGAN are closer to those of the original high-resolution images than to those obtained with any state-of-the-art
method.
@InProceedings{srgan,
author = {Christian Ledig, Lucas Theis, Ferenc Huszar, Jose Caballero, Andrew Cunningham, Alejandro Acosta, Andrew Aitken, Alykhan Tejani, Johannes Totz, Zehan Wang, Wenzhe Shi},
title = {Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial Network},
booktitle = {arXiv},
year = {2016}
}