#計算圖片的PSNR和SSIM代碼示例
import os
import CV2
from skimage.metrics import structural_similarity
from skimage.metrics import mean_squared_error
from skimage.metrics import peak_signal_noise_ratio

#設(shè)置保存結(jié)果的文件
out1 = './PSNR.txt'
out2 = './SSIM.txt'
#以可寫方式創(chuàng)建文本文件
f1 = open(out1,'w') 
f2 = open(out2,'w')
#原始數(shù)據(jù)的路徑文件夾
dir1 = './dir00'


files = os.listdir(dir1)
t1 = CV2.imread('00002.png')

#這里的file對應(yīng)dir1文件夾中所有數(shù)據(jù)的名稱
for file in files:
 ?	 #os.path.join(dir1,file) 表示將dir1與file兩個路徑名連接，表示當(dāng)前數(shù)據(jù)的路徑
 ? ?t2 = CV2.imread(os.path.join(dir1,file))
 ? ?PSNR = peak_signal_noise_ratio(t1, t2)
 ? ?#PSNRs.append(PSNR)
 ? ?SSIM = structural_similarity(t1, t2, multichannel=True) #channel_axis
 ? ?f1.writelines('%s: %.6f\n'%(file,PSNR))
 ? ?f2.writelines('%s: %.6f\n'%(file,SSIM))
 ? ?#SSIMs.append(SSIM)
f1.close()
f2.close() ? ?