Skip to content

Fid¤

Module: generative_models.core.evaluation.metrics.image.fid

Source: generative_models/core/evaluation/metrics/image/fid.py

Overview¤

Fréchet Inception Distance (FID) implementation using JAX and NNX.

FID is a metric that calculates the distance between feature vectors of real and generated images. It uses the Inception-v3 network to extract features and then computes the Fréchet distance between distributions.

Classes¤

FrechetInceptionDistance¤

class FrechetInceptionDistance

Functions¤

init¤

def __init__()

calculate_frechet_distance¤

def calculate_frechet_distance()

compute¤

def compute()

feature_extractor¤

def feature_extractor()

Module Statistics¤

  • Classes: 1
  • Functions: 4
  • Imports: 6