(10-3)基于多模态模型的文生图系统:多模态成对抗网络(GAN)模型(2)

(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

猜你喜欢

转载自blog.csdn.net/asd343442/article/details/143489553