Skip to content
Pytorch implementation of "SinGAN: Learning a Generative Model from a Single Natural Image"
Branch: master
Clone or download
FriedRonaldo Merge pull request #2 from arshamg/master
Remove unused import (scipy.misc import imread)
Latest commit f644b0d Nov 11, 2019
Type Name Latest commit message Commit time
Failed to load latest commit information.
code Remove unused import Nov 6, 2019
src Add images Jul 21, 2019 Update Sep 26, 2019

SinGAN: Learning a Generative Model from a Single Natural Image

Pytorch implementation of "SinGAN: Learning a Generative Model from a Single Natural Image" (arxiv)

Official repository : SinGAN Official Pytorch implementation

This implementation is based on these repos.


This repository is not official implementation. The official one is here : SinGAN Official Pytorch implementation.



We introduce SinGAN, an unconditional generative model that can be learned from a single natural image. Our model is trained to capture the internal distribution of patches within the image, and is then able to generate high quality, diverse samples that carry the same visual content as the image. SinGAN contains a pyramid of fully convolu- tional GANs, each responsible for learning the patch distri- bution at a different scale of the image. This allows generat- ing new samples of arbitrary size and aspect ratio, that have significant variability, yet maintain both the global struc- ture and the fine textures of the training image. In contrast to previous single image GAN schemes, our approach is not limited to texture images, and is not conditional (i.e. it gen- erates samples from noise). User studies confirm that the generated samples are commonly confused to be real im- ages. We illustrate the utility of SinGAN in a wide range of image manipulation tasks.


  • Multi-scale GAN with progression
  • Initialization via copy
  • Scaling noise by the root mean square error between input image and reconstructed one
  • Zero padding at the image level (not feature level)
  • WGAN-GP loss

Additional implementation for better quality

  • LSGAN loss
  • Non-saturating loss with zero-centered gradient penalty


  • GAN with Zero-centered GP and larger weight of reconstruction loss exhibits better quality.
  • Inter-run variance is large


  • python 3.6
  • pytorch 1.0.0 or 1.1.0
  • torchvision 0.2.2 or 0.3.0
  • tqdm
  • scipy
  • PIL
  • opencv-python (cv2)

Data Preparation

|--- data
|    |--- SinGANdata
|             |--- trainPhoto
|             |--- testPhoto
|--- SinGAN
     |--- models
     |        |---
     |        |--- ...
     | ...
  • Then, an image in "trainPhoto" will be selected randomly for training.

How to Run


  • data_dir
    • Path of dataset. If you follow the example hierarchy, let it default.
  • dataset
    • Dataset to use. It is fixed to "PHOTO".
  • gantype
    • Loss type of GANs. You can choose among "wgangp, zerogp, lsgan". Recommend to use "zerogp".
  • model_name
    • Prefix of the directory of log files.
  • workers
    • Workers to use for loading dataset.
  • batch_size
    • Size of batch, it is fixed to "1". SinGAN uses only one image.
  • val_batch
    • Size of batch in validation, it is fixed to "1". SinGAN uses only one image.
  • img_size_max
    • Size of largest image. = Finest
  • img_size_min
    • Size of smallest image. = Coarsest
  • img_to_use
    • Index of the image to use. If you do not change, it will be sampled randomly.
  • load_model
    • Directory of the model to load.
  • validation
    • If you call, the code will go into the validation mode.
  • test
    • If you call, the code will go into the test mode. But, it is the same as validation essentially.
  • world-size
    • Do not change.
  • rank
    • Do not change.
  • gpu
    • GPU number to use. You should set. Unless it utilizes all the available GPUs.
  • multiprocessing-distributed
    • Do not use in this setting.
  • port
    • Do not use in the setting.

SinGAN uses only one image to train and test. Therefore multi-gpus mode is not supported.


  • Use NS loss with zero-cented GP and 0-th gpu. The train image will be selected randomly. It will generate (1052, 1052) images at last.
python --gpu 0 --gantype zerogp --img_size_max 1025
  • Use WGAN-GP loss to train and 0-th gpu.
python --gpu 0 --img_to_use 0 --img_size_max 1025 --gantype wgangp


python --gpu 0 --img_to_use 0 --img_size_max 1025 --gantype zerogp --validation --load_model $(dir)
  • Running the script in "Train" stores the intermediate results automatically. So, you do not need to run "Test".


  • Thunder image is from : Google

  • Original / Reconstructed / Generated (33, 59, 105, 187 px)
    org rec gen
    org rec gen
    org rec gen
    org rec gen

  • More generation results

    • 187px / Last two images are recon/real, respectively.
      gen gen gen
      gen gen gen
      gen gen gen
      gen gen gen
    • 187px / Recon. image and real one are placed among fakes.
      gen gen gen
      gen gen gen
      gen gen gen
      gen gen gen
    • 247px
      gen gen gen
      gen gen gen
      gen gen gen
      gen gen gen
    • 591px
      gen gen
      gen gen
      gen gen


  • Shaham, Tamar Rott, Tali Dekel, and Tomer Michaeli. "SinGAN: Learning a Generative Model from a Single Natural Image." arXiv preprint arXiv:1905.01164 (2019).
  • Gulrajani, Ishaan, et al. "Improved training of wasserstein gans." Advances in neural information processing systems. 2017.
  • Mescheder, Lars, Andreas Geiger, and Sebastian Nowozin. "Which training methods for GANs do actually converge?." arXiv preprint arXiv:1801.04406 (2018).
  • Zhu, Jun-Yan, et al. "Unpaired image-to-image translation using cycle-consistent adversarial networks." Proceedings of the IEEE international conference on computer vision. 2017.
You can’t perform that action at this time.