LPIPS图像相似性度量标准:The Unreasonable Effectiveness of Deep Features as a Perceptual Metric
目录
一、感知相似性
人类可以快速评估两幅图像之间的感知相似性,但是底层过程非常复杂。
纹理图像包含了纹理颜色、纹理基元等丰富的图像信息。在计算机视觉研究领域中,人们使用感知
相似性来度量不同纹理之间的相似程度,研究人类对纹理图像的视觉感知。纹理相似性度量广泛应用于纹理识别和材质识别,是对象识别和场景理解的关键技术之一。研究人员通过计算特征之间的距离度量估计纹理感知相似性。
近年来,对图像网络分类进行训练的VGG网络的特性作为图像合成的训练损失具有显著的作用。
二、传统度量和深度学习法
将左右的两个图像块和中间的图像块进行比较:
如图表示,每一组有三张图片,由传统的评价标准如L2、SSIM、PSNR等评价结果和人体认为的大不相同,这是传统方法的弊端。如果图片平滑,那么传统的评价方式则大概率会失效。而目前GAN尤其是VAE等生成模型生成结果都过于平滑。
而最后三行的评价为深度学习的方式,可以看到,通过神经网络(非监督、自监督、监督模型)提取特征的方式,并对特征差异进行计算能够有效进行评价,而且能够和人体评价相似。
三、原理
如图所示,将两个输入送入神经网络F(可以为VGG、Alexnet、Squeezenet)中进行特征提取,对每个层的输出进行激活后归一化处理,记为,然后经过w层权重点乘后计算L2距离(有小伙伴指出,在源码中,并未进行开方,可以查看代码122-135行,多谢指点,谢谢!https://github.com/richzhang/PerceptualSimilarity/blob/31bc1271ae6f13b7e281b9959ac24a5e8f2ed522/lpips/lpips.py#L122https://github.com/richzhang/PerceptualSimilarity/blob/31bc1271ae6f13b7e281b9959ac24a5e8f2ed522/lpips/lpips.py#L122),最后取平均获得距离。论文将该度量记为LPIPS。
公式如下:(因此实际中,公式不开方)
当w值为1时,即公式计算余弦距离。
论文中将LPIPS分为三类:
四、测试和代码
1、python环境下安装 pip install lpips
2、准备好图片,注意:图片读取后应归一化,且注意数据类型,见代码。我的图片如下所示,为1536*512大小。你可以直接载入你自己的图片。我们比较的是模型结果和目标结果。
3、代码:(注意可以选择预训练网络模型为alex或者vgg,这里我选择的是alex)
#coding=gbk
import cv2
import lpips
import torchvision.transforms as transforms
import torch
#载入指标模型
loss_fn_alex = lpips.LPIPS(net='alex') # best forward scores
loss_fn_vgg = lpips.LPIPS(net='vgg') # closer to "traditional" perceptual loss, when used for optimization
#读取图片
test1 = cv2.imread('test_picture_117.png')
test2 = cv2.imread('test_picture_124.png')
#分割目标图片
test1_org = test1[:,:512,:]/255 #原始图片
test1_res = test1[:,512:1024,:]/255 #模型输出结果
test1_label = test1[:,1024:1536,:]/255 #label图片
test2_org = test2[:,:512,:]/255
test2_res = test2[:,512:1024,:]/255
test2_label = test2[:,1024:1536,:]/255
#转为tensor
transf = transforms.ToTensor()
test1_org = transf(test1_org)
test1_res = transf(test1_res)
test1_label = transf(test1_label)
test2_org = transf(test2_org)
test2_res = transf(test2_res)
test2_label = transf(test2_label)
#转换数据类型
test1_orgg = test1_org.to(torch.float32)
test1_ress = test1_res.to(torch.float32)
test1_labell = test1_label.to(torch.float32)
test2_orgg = test2_org.to(torch.float32)
test2_ress = test2_res.to(torch.float32)
test2_labell =test2_label.to(torch.float32)
#测试
d11 = loss_fn_alex(test1_ress, test1_labell)
d12 = loss_fn_alex(test1_ress, test2_labell)
print('d11:',d11)
print('d12:',d12)
d22 = loss_fn_alex(test2_ress, test2_labell)
d21 = loss_fn_alex(test2_ress, test1_labell)
print('d22:',d22)
print('d121:',d21)
五、测试结果
值越小越好。d11、d22表示生成结果和对应的label之间的LPIPS距离,d12、d21为生成结果和非对应label结果。可见对应的情况下值应当低且实际结果如下所示,事实确实如此。
六、参考文献
更多推荐
所有评论(0)