當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


Python tf.image.psnr用法及代碼示例


返回 a 和 b 之間的峰值 Signal-to-Noise 比率。

用法

tf.image.psnr(
    a, b, max_val, name=None
)

參數

  • a 第一組圖像。
  • b 第二組圖像。
  • max_val 圖像的動態範圍(即最大值和最小值之間的差異)。
  • name 將計算嵌入的命名空間。

返回

  • a 和 b 之間的標量 PSNR。返回的張量具有類型 tf.float32 和形狀 [batch_size, 1]。

這旨在用於信號(或圖像)。批量為每個圖像生成一個 PSNR 值。

輸入的最後三個維度預計為[高度、寬度、深度]。

例子:

# Read images from file.
    im1 = tf.decode_png('path/to/im1.png')
    im2 = tf.decode_png('path/to/im2.png')
    # Compute PSNR over tf.uint8 Tensors.
    psnr1 = tf.image.psnr(im1, im2, max_val=255)

    # Compute PSNR over tf.float32 Tensors.
    im1 = tf.image.convert_image_dtype(im1, tf.float32)
    im2 = tf.image.convert_image_dtype(im2, tf.float32)
    psnr2 = tf.image.psnr(im1, im2, max_val=1.0)
    # psnr1 and psnr2 both have type tf.float32 and are almost equal.

相關用法


注:本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.image.psnr。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。