计算LPIPS
import lpipsimport osimport utils_image as utilloss_fn_alex = lpips.LPIPS(net='vgg')#也可以选择alexinputpath = r'E:\testresult'#(alex:0.1009) (vgg:0.1879)gtpath=r'F:\eval_normal'imgs=os.listdir(path)losssu
·
import lpips
import os
import utils_image as util
loss_fn_alex = lpips.LPIPS(net='vgg')#也可以选择alex
inputpath = r'E:\testresult'#(alex:0.1009) (vgg:0.1879)
gtpath=r'F:\eval_normal'
imgs=os.listdir(path)
losssum=0.0
count=0
for img in imgs:
ig=util.imread_uint(inputpath +'/'+img,3)
input=util.uint2tensor3(ig)
#print(path+'/'+img,pathgt + '/normal' + img[3:])
gt = util.imread_uint(gtpath+ '/' + img, 3)
#gt = util.imread_uint(pathgt + '/normal' + img[3:], 3)
input = util.uint2tensor3(ig)
gt = util.uint2tensor3(gt)
loss=loss_fn_alex(input,gt)
losssum=losssum+loss
count+=1
print(img,loss.item())
print(losssum/count)
DAMO开发者矩阵，由阿里巴巴达摩院和中国互联网协会联合发起，致力于探讨最前沿的技术趋势与应用成果，搭建高质量的交流与分享平台，推动技术创新与产业应用链接，围绕“人工智能与新型计算”构建开放共享的开发者生态。
更多推荐
1
0- 0
已为社区贡献1条内容
所有评论(0)