Skip to content

An empirical study on evaluation metrics of generative adversarial networks.

Notifications You must be signed in to change notification settings

xuqiantong/GAN-Metrics

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

32 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

Contributors 4

  •  
  •  
  •  
  •