mseitzer/pytorch-fid
Compute FID scores with PyTorch.
Ports the official TensorFlow FID implementation with PyTorch, computing Fréchet distance between Gaussian-fitted Inception network features to measure image dataset similarity. Supports flexible feature extraction from intermediate Inception layers (64–2048 dimensions) via global average pooling, enabling evaluation on smaller datasets, and can pre-compute and cache dataset statistics as `.npz` archives for efficient multi-model comparisons.
3,833 stars and 112,753 monthly downloads. No commits in the last 6 months. Available on PyPI.
Stars
3,833
Forks
529
Language
Python
License
Apache-2.0
Last pushed
Jul 03, 2024
Monthly downloads
112,753
Commits (30d)
0
Dependencies
5
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/mseitzer/pytorch-fid"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Related frameworks
Naresh1318/Adversarial_Autoencoder
A wizard's guide to Adversarial Autoencoders
acids-ircam/RAVE
Official implementation of the RAVE model: a Realtime Audio Variational autoEncoder
ratschlab/aestetik
AESTETIK: Convolutional autoencoder for learning spot representations from spatial...
jaanli/variational-autoencoder
Variational autoencoder implemented in tensorflow and pytorch (including inverse autoregressive flow)
nathanhubens/Autoencoders
Implementation of simple autoencoders networks with Keras