torch_fidelity_is_fid_kid_metrics_calculation.py

1import torch_fidelity
2
3# Metrics calculation for a generative model
4# Replace 'MyGenerativeModel' with your actual model class
5# and 'my_input_dataset' with your dataset or data loader
6metrics = torch_fidelity.calculate_metrics(
7    input1=torch_fidelity.GenerativeModelModuleWrapper(my_model, 128, 'normal', 0),
8    input1_model_num_samples=10000,
9    input2='cifar10-train',
10    cuda=True,
11    isc=True,
12    fid=True,
13    kid=True,
14    verbose=True,
15)
16
17print(f"Inception Score: {metrics['inception_score_mean']} ± {metrics['inception_score_std']}")
18print(f"Frechet Inception Distance: {metrics['frechet_inception_distance']}")
19print(f"Kernel Inception Distance: {metrics['kernel_inception_distance_mean']} ± {metrics['kernel_inception_distance_std']}")