遍历图像对计算psnr,ssim,IE

最新推荐文章于 2024-07-19 19:08:28 发布

MoonShapedPool

最新推荐文章于 2024-07-19 19:08:28 发布

阅读量776

点赞数 1

分类专栏： python 文章标签： python

本文链接：https://blog.csdn.net/MoonShapedPool/article/details/89359922

版权

python 专栏收录该内容

6 篇文章 0 订阅

订阅专栏

import os
import numpy as np
from scipy.misc import imread ,imresize
from PIL import Image
from skimage import color
from skimage.measure import compare_ssim, compare_psnr, compare_mse
from tqdm import tqdm
path1 = './A'
path2 = './B'

total_p = 0
total_s = 0

print('[***]calc...')
images = os.listdir(path1)
for image in tqdm(images):
    # image pair have same size

    img1 = imresize(imread(os.path.join(path1,image)),(576,720))
    # print(type(img1)) = numpy.ndarray
    img2 = imread(os.path.join(path2,'B'+image[1:]))
    # 断言尺寸不同则抛出异常    
    assert img1.size == img2.size ,\
	"image pair should have same size,check the img_size(w,h) "

    err = img1 - img2
    ie = np.mean(np.sqrt(np.sum(err * err, axis=2)))

    psnr = compare_psnr(img1,img2)
    ssim = compare_ssim(img1,img2,multichannel=1)
    n = len(images)
    # print(psnr,ssim)
    total_p += psnr
    total_s += ssim
    mean_p = total_p/n
    mean_s = total_s/n

print("avg_PSNR , avg_SSIM :","%.2f"% mean_p,",","%.3f"% mean_s)
print('Interpolation Error :',"%.2f"% ie)