Skip to content

PyTorch implementation of 1712.06087 "Zero-Shot" Super-Resolution using Deep Internal Learning

License

Notifications You must be signed in to change notification settings

jacobgil/pytorch-zssr

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Unofficial PyTorch implementation of "Zero-Shot" Super-Resolution using Deep Internal Learning

Unofficial Implementation of 1712.06087 "Zero-Shot" Super-Resolution using Deep Internal Learning by Assaf Shocher, Nadav Cohen, Michal Irani.

Official Project page: http://www.wisdom.weizmann.ac.il/~vision/zssr/

Paper: https://arxiv.org/abs/1712.06087


This trains a deep neural network to perform super resolution using a single image.

The network is not trained on additional images, and only uses information from within the target image. Pairs of high resolution and low resolution patches are sampled from the image, and the network fits their difference.

Low resolution ZSSR

ZSSR ZSSR


TODO:

  • Implement additional augmentation using the "Geometric self ensemble" mentioned in the paper.
  • Implement gradual increase of the super resolution factor as described in the paper.
  • Support for arbitrary kernel estimation and sampling with arbitrary kernels. The current implementation interpolates the images bicubic interpolation.

Deviations from paper:

  • Instead of fitting the loss and analyzing it's standard deviation, the network is trained for a constant number of batches. The learning rate shrinks x10 every 10,000 iterations.

Usage

Example: python train.py --img img.png

usage: train.py [-h] [--num_batches NUM_BATCHES] [--crop CROP] [--lr LR]
                [--factor FACTOR] [--img IMG]

optional arguments:
  -h, --help            show this help message and exit
  --num_batches NUM_BATCHES
                        Number of batches to run
  --crop CROP           Random crop size
  --lr LR               Base learning rate for Adam
  --factor FACTOR       Interpolation factor.
  --img IMG             Path to input img

About

PyTorch implementation of 1712.06087 "Zero-Shot" Super-Resolution using Deep Internal Learning

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages