参考博客Skimage.measure 的compare_psnr和compare_ssim的改动

from skimage.metrics import structural_similarity as compare_ssim
from skimage.metrics import peak_signal_noise_ratio as compare_psnr

import cv2

img1 = cv2.imread(r'HanJie-20220524-102819-Ping-10260.jpg')
img2 = cv2.imread(r'result.jpg')

psnr = compare_psnr(img1, img2)
ssim = compare_ssim(img1, img2, multichannel=True)  # 对于多通道图像(RGB、HSV等)关键词multichannel要设置为True
#mse = compare_mse(img1, img2)

print('PSNR:{},SSIM:{}'.format(psnr, ssim))

运行结果:
在这里插入图片描述

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐