qq_42443342的博客

原创 pytorch实现FID分数计算

FID分数计算需要先将图片输入到预训练的inception_v3模型中，我输入的图片通道数为1，将inception_v3第一层的卷积通道数改为1。此外，计算FID分数，我们需要inception_v3模型输出的2048维度特征矩阵，需要删除最后一层全连接层。inception_v3模型输入图片要求最小是75*75，最佳是299*299。用numpy中的mean()和np.cov()计算特征均值和协方差矩阵，在计算特征时利用numpy()将输出转化为nd.array。根据FID公式计算FID分数。