pip install pytorch-fid

from pytorch_fid import fid_score# 准备真实数据分布和生成模型的图像数据
real_images_folder = 'xxx'
generated_images_folder = 'xxx'# 计算FID距离值
fid_value = fid_score.calculate_fid_given_paths([real_images_folder, generated_images_folder],device="cuda",batch_size=1,dims=2048)
print('FID value:', fid_value)

注意：scipy需要安装低版本

pip install scipy==1.11

参考文章

【pytorch】FID讲解以及pytorch实现_pytorch-fid-CSDN博客

mseitzer/pytorch-fid: Compute FID scores with PyTorch. (github.com)