前言:对于生成式模型来说,如何评价是一个难题,特别是对于GANs这种容易坍塌的模型来说,选择合适的度量指标是关键。IS和FID是最常用的两种指标,都依赖于Inception模型。本文介绍如何利用tensorflow/pytorch官方提供的预训练Inception V3模型,五分钟内搭建自己的IS和FID计算代码。
目录
IS简介
IS基于Google的预训练网络Inception Net-V3,Inception Net-V3是精心设计的卷积网络模型,输入为图片张量,输出为1000维向量。输出向量的每个维度的值对应图片属于某类的概率,因此整个向量可以看做一个概率分布。
p(y|x)表示Inception输入生成图像x时的输出分布,P(x)表示生成器G生成图像x的概率,P(yi|x)表示Inception预测x为第i类的概率,IS是衡量两者之间的KL散度:
IS越大,生成图片的质量越高。