kam1107/RealnessGAN
Code for ICLR2020 paper ‘Real or Not Real, that is the Question’
repo name | kam1107/RealnessGAN |
repo link | https://github.com/kam1107/RealnessGAN |
homepage | |
language | Python |
size (curr.) | 17004 kB |
stars (curr.) | 151 |
created | 2020-01-21 |
license | |
RealnessGAN
This repository contains the code of the following paper:
Real or Not Real, that is the Question Yuanbo Xiangli*, Yubin Deng*, Bo Dai*, Chen Change Loy, Dahua Lin https://openreview.net/forum?id=B1lPaCNtPB
Abstract: While generative adversarial networks (GAN) have been widely adopted in various topics, in this paper we generalize the standard GAN to a new perspective by treating realness as a random variable that can be estimated from multiple angles. In this generalized framework, referred to as RealnessGAN, the discriminator outputs a distribution as the measure of realness. While RealnessGAN shares similar theoretical guarantees with the standard GAN, it provides more insights on adversarial learning. More importantly, compared to multiple baselines, RealnessGAN provides stronger guidance for the generator, achieving improvements on both synthetic and real-world datasets. Moreover, it enables the basic DCGAN architecture to generate realistic images at 10241024 resolution when trained from scratch.*
Dataset
Experiments were conducted on two real-world datasets: CelebA and FFHQ; and a toy dataset: Mixture of Gaussians.
Requirements
- Python 3.6
- Pytorch (Latest from source)
Training
- Either use the aforementioned dataset or prepare your dataset.
- Scripts to run experiments are stored in /scripts/*.sh.
- Edit folder locations in your scripts. Make sure the folders to store LOG and OUTPUT are created.
- Run
./scripts/run_your_scripts.sh
Snapshots
CelebA 256x256 (FID = 23.51)
FFHQ 1024x1024 (FID = 17.18)