两幅图像的相似度——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("_______________________________________")