欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

两幅图像的相似度——psnr计算

程序员文章站 2024-03-26 13:42:29
...

W3S-TensorFlow函数教程
两幅图像的相似度——psnr计算
缘由:
高分辨率原图 和 GAN网络由低分辨率图像生成的高分辨图像的 psnr 计算
分析:
使用 tf 读取图片、解码数据、带入tf.image.psnr计算
图片读取参考链接

import tensorflow as tf

 # Read images from file.
        gen_path = “生成的图像路径”
        hr_path = “高分辨率原图路径”

        gen_img_data = tf.gfile.FastGFile(gen_path, "rb").read()
        hr_img_data = tf.gfile.FastGFile(hr_path, "rb").read()

        gen_img = tf.image.decode_png(gen_img_data)
        hr_img = tf.image.decode_png(hr_img_data)

        # Compute PSNR over tf.uint8 Tensors.
        psnr1 = tf.image.psnr(gen_img, hr_img, max_val=255.)

        print(psnr1)
        with tf.Session() as sess:
            psnr = psnr1.eval()
            print(psnr)
        print("_______________________________________")