(7)函数calculate_FID_CLIP_sim的功能是计算生成器产生的图像的FID(Fréchet Inception Distance)和CLIP相似度评分。它使用InceptionV3模型计算FID,并使用CLIP模型计算图像和文本之间的相似度评分。
def calculate_FID_CLIP_sim(dataloader, text_encoder, netG, CLIP, device, m1, s1, epoch, max_epoch, times, z_dim,
batch_size):
"""
计算FID(Fréchet Inception Distance)和CLIP相似度
"""
# 初始化CLIP相似度
clip_cos = torch.FloatTensor([0.0]).to(device)
# 设置用于FID的InceptionV3模型
dims = 2048
block_idx = Incept