Jun, 2020

基于 Watson 感知模型的生成神经网络的损失函数

TL;DR为了训练变分自编码器以生成真实图像,我们提出了一种基于 Watson 视觉模型的损失函数,通过计算颜色图像中的加权距离和掩模来反映人类对图像相似性的知觉,并将其拓展到彩色图像上。在实验中,VAE 利用新的损失函数进行训练可生成更真实、高质量的图像样本,相较于欧几里得距离和结构相似性指数,图像模糊度更低,与基于深度神经网络的损失相比,新方法需要更少的计算资源,并且生成的图像具有更少的伪影。