torch_fidelity_fid_is_kid_metrics_calculation.py

1import torch
2from torch_fidelity import calculate_metrics
3
4# Replace these with your actual model and dataset
5# In this example, we assume a generative model `generator` and a real dataset loader `dataloader`
6# are already defined. 
7
8# Example: Calculate FID, IS, and Kernel Inception Distance (KID)
9metrics = calculate_metrics(
10    input1="path/to/generated_images_folder", 
11    input2="path/to/real_images_folder", 
12    cuda=True, 
13    isc=True, 
14    fid=True, 
15    kid=True, 
16    verbose=True
17)
18
19print(f"Inception Score: {metrics['inception_score_mean']}")
20print(f"Frechet Inception Distance: {metrics['frechet_inception_distance']}")
21print(f"Kernel Inception Distance: {metrics['kernel_inception_distance_mean']}")