kozistr / gan-metrics

Lots of evaluation metrics for the generative adversarial networks in pytorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

gan-metrics

Lots of evaluation metrics of Generative Adversarial Networks in pytorch

Work In Progress...

Requirements

  • Python 3.x
  • torch 1.x
  • torchvision 0.4.x
  • numpy
  • scipy
  • cython
  • pot
  • pyyaml

Usage

  1. Clone the repository and Move the directory.
$ git clone https://github.com/kozistr/gan-metrics
$ cd ./gan-metrics
  1. Install the requirements.txt
$ pip3 install -r requirements.txt
  1. Configure the config.yml
open the file and edit
  1. Run
$ python3 -m metrics

Metrics

Metric Paper Code
Inception Score (IS) arXiv code
Frechet Inception Distance (FID) code
Kernel Inception Distance (KID)
SSIM & PSNR
Precision & Recall
Earth Mover's Distance (EMD, a.k.a wasserstein distance) wiki code
Perceptual Path Length (PPL)
Learned Perceptual Image Patch Similarity (LPIPS) arXiv
Reliable Fidelity & Diversity (PRDC) arXiv
Amazon's Mechanical Turk (AMT) Amazon

DataSets

You can edit the transform function with your flavor!

  • MNIST / FashionMNIST
  • CIFAR10 / CIFAR100
  • LSUN
  • CelebA
  • ImageNet
  • Custom

References

Author

Hyeongchan Kim / @kozistr

About

Lots of evaluation metrics for the generative adversarial networks in pytorch

License:Apache License 2.0


Languages

Language:Python 98.8%Language:Makefile 1.2%