torch-fidelity 0.3.0


pip install torch-fidelity

  Latest version

Released: Jun 15, 2021

Project Links

Meta
Author: Anton Obukhov
Requires Python: >=3.6

Classifiers

Evaluation of generative models such as GANs is an important part of the deep learning research.
In the domain of 2D image generation, three approaches became widely spread: Inception Score
(aka IS), Fréchet Inception Distance (aka FID), and Kernel Inception Distance (aka KID).

These metrics, despite having a clear mathematical and algorithmic description, were initially
implemented in TensorFlow, and inherited a few properties of the framework itself and the code
they relied upon. These design decisions were effectively baked into the evaluation protocol and
became an inherent part of the metrics specification. As a result, researchers wishing to
compare against state of the art in generative modeling are forced to perform evaluation using
codebases of the original metric authors. Reimplementations of metrics in PyTorch and other
frameworks exist, but they do not provide a proper level of fidelity, thus making them
unsuitable for reporting results and comparing to other methods.

This software aims to provide epsilon-exact implementations of the said metrics in PyTorch, and thus
remove inconveniences associated with generative models evaluation and development.
Find more details and the most up-to-date information on the project webpage:
https://www.github.com/toshas/torch-fidelity


Wheel compatibility matrix

Platform Python 3
any

Files in release

Extras: None
Dependencies:
numpy
Pillow
scipy
torch
torchvision
tqdm