Awesome
SinGAN
This is an unofficial implementation of SinGAN from someone who's been sitting right next to SinGAN's creator for almost five years.
<p align="center"> <img width="992" height="372" src="/figures/sampled.png"> </p>Please refer the project's page for more details.
Citation
If you use this code for your research, please cite the paper:
@inproceedings{shaham2019singan,
title={Singan: Learning a generative model from a single natural image},
author={Shaham, Tamar Rott and Dekel, Tali and Michaeli, Tomer},
booktitle={Proceedings of the IEEE/CVF International Conference on Computer Vision},
pages={4570--4580},
year={2019}
}
Code
Clone repository
Clone this repository into any place you want.
git clone https://github.com/kligvasser/SinGAN
cd ./SinGAN/generation/
Install dependencies
python -m pip install -r requirements.txt
This code tested in PyTorch 1.8.1.
Training
To train SinGAN model on your own image:
python3 main.py --root <path-to-image>
Evaluating
For evaluating, run the following command:
python3 main.py --root <path-to-image> --evaluation --model-to-load <path-to-model-pt> --amps-to-load <path-to-amp-pt> --num-steps <number-of-samples> --batch-size <number-of-images-in-batch>