Python日常记录

#计算图片的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

#设置保存结果的文件
out1 = './PSNR.txt'
out2 = './SSIM.txt'
#以可写方式创建文本文件
f1 = open(out1,'w') 
f2 = open(out2,'w')
#原始数据的路径文件夹
dir1 = './dir00'


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

#这里的file对应dir1文件夹中所有数据的名称
for file in files:
  	 #os.path.join(dir1,file) 表示将dir1与file两个路径名连接，表示当前数据的路径
    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()