Skip to content
Pytorch implementation of OCFGAN-GP (CVPR 2020, Oral).
Python Shell
Branch: master
Clone or download

Latest commit

Fetching latest commit…
Cannot retrieve the latest commit at this time.


Type Name Latest commit message Commit time
Failed to load latest commit information.
data Added code Mar 14, 2020
src Added code Mar 14, 2020
.gitignore Added code Mar 14, 2020
LICENSE Initial commit Mar 10, 2020 Update Mar 14, 2020 Added code Mar 14, 2020 Added code Mar 14, 2020 Added code Mar 14, 2020 Added code Mar 14, 2020 Added code Mar 14, 2020 Added code Mar 14, 2020


This repository contains code for the CVPR 2020 paper (Oral) A Characteristic Function Approach to Deep Implicit Generative Modeling. In this paper, we propose a method of training a Generative Adversarial Network (GAN) by minimizing the expected distance between empirical characteristic functions of the real and generated data distributions. The generator seeks to minimize what we term as the Empirical Characteristic Function Distance (ECFD), whereas the discriminator maximizes it. Given samples from a distribution, the empirical characteristic function is given by

and the ECFD between two distributions is given by

Fig. 1: Samples generated by OCFGAN-GP trained on CelebA 128 x 128 dataset.


  • Pytorch (tested on v1.2.0) & torchvision (tested on v0.4.0)
  • Numpy (tested on v1.17.2)
  • OpenCV (tested on v4.1.1)
  • Scipy
  • Pillow
  • tqdm
  • scikit-learn
  • CUDA 10


Generating samples from pre-trained models

  • Download the pre-trained generators from releases.
  • Run the following command to generate an 8x8 grid of samples from a model trained on CIFAR10 dataset:
python src/\
 --imsize 32\
 --noise_dim 32\
 --gen flexible-dcgan\
 --out_dir ./ checkpoint.pth
  • Refer to the shell scripts for --imsize, --noise_dim, and --gen parameters for other datasets.

Training your own model

  • Downloading Datasets: All the datasets will download by themselves when the code is run, except CelebA. CelebA can be downloaded by executing python celebA. Rename the directory ./data/img_align_celeba to ./data/celebA after the script finishes execution.
  • Run python src/ --help to see a description of all the available command-line arguments.
  • Example: run the following command to train on the CIFAR10 dataset:
python src/\
 --dataset cifar10\
 --dataroot ./data/cifar10\
 --model cfgangp\
 --batch_size 64\
 --image_size 32\
 --nc 3\
 --noise_dim 32\
 --dout_dim 32\
 --max_giter 125000\
 --resultsroot ./out
 --gpu_device 0\
 --num_freqs 8\
 --weight gaussian_ecfd\
 --sigmas 0.
  • Alternatively, use the provided shell scripts to train the models. For example, run bash to train OCFGAN-GP on the STL10 dataset.

The generated images will be written in ./out/<dataset>/<model>/<gen-network>/samples/ and the model checkpoints will be saved in ./out/<dataset>/<model>/<gen-network>/checkpoints/.


For any questions regarding the code or the paper, please email me at


  title={A Characteristic Function Approach to Deep Implicit Generative Modeling},
  author={Ansari, Abdul Fatir and Scarlett, Jonathan and Soh, Harold},
  booktitle={IEEE Conference on Computer Vision and Pattern Recognition},


Parts of the code/network structures in this repository have been adapted from the following repos:

You can’t perform that action at this time.