Lightning-AI / torchmetrics

Torchmetrics - Machine learning metrics for distributed, scalable PyTorch applications.
https://lightning.ai/docs/torchmetrics/
Apache License 2.0
2.08k stars 398 forks source link

Add support for the FID metric. #169

Closed ohayonguy closed 3 years ago

ohayonguy commented 3 years ago

🚀 Feature

Add support for the Frechet Inception Distance as a metric, to evaluate generative models (https://arxiv.org/abs/1706.08500)

Motivation

This metric is used in many recent GAN applications to evaluate the perceptual quality of the generated images. It is therefore a very helpful and important metric to implement.

Alternatives

The inception score could be also used as well, but FID is more acceptable today.

github-actions[bot] commented 3 years ago

Hi! thanks for your contribution!, great first issue!

Borda commented 3 years ago

@ohayonguy are you interested in implementing it? closing this one in favor of #65 let's continue the discussion there...