Skip to content

An empirical study on evaluation metrics of generative adversarial networks.

Notifications You must be signed in to change notification settings

limingyao001/GAN-Metrics

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

30 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

GAN Metrics

This repository provides the code for An empirical study on evaluation metrics of generative adversarial networks.

Requirement

  • Python 3.6.4
  • torch 0.4.0
  • torchvision 0.2.1
  • pot 0.4.0
  • tqdm 4.19.6
  • numpy, scipy, math

Usage

  • We create a demo for DCGAN training as well as computing all the metrics after each epoch.
    In the demo, final metrics scores of all epoches will be scored in <outf>/score_tr_ep.npy
  • If you want to compute metrics of your own images, you have to modify the codes of function compute_score_raw() in metric.py by yourself :)
python3 demo_dcgan.py \
--dataset cifar10 \
--cuda \
--dataroot <data_folder> \
--outf <output_folder> \
--sampleSize 2000

demo

About

An empirical study on evaluation metrics of generative adversarial networks.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 91.0%
  • Python 9.0%