图像质量评价指标之 PSNR 和 SSIM

原理:https://zhuanlan.zhihu.com/p/50757421

'''
compute PSNR with tensorflow
'''
import tensorflow as tf


def read_img(path):
    return tf.image.decode_image(tf.read_file(path))


def psnr(tf_img1, tf_img2):
    return tf.image.psnr(tf_img1, tf_img2, max_val=255)


def _main():
    # t1 = read_img('./Data/00001_00_0.1s_gt.png')               #Clean Image
    # t2 = read_img('./Data/00001_00_0.1s_out_optimized.png')       #Noise Image
    t1 = read_img('./result_Sony_1/final/10003_00_100_out.png')
    t2 = read_img('./result_Sony_1/final/10003_00_100_gt.png')
    with tf.Session() as sess:
        sess.run(tf.global_variables_initializer())
        y = sess.run(psnr(t1, t2))
        print(y)


if __name__ == '__main__':
    _main()