tf.image.psnr

Возвращает пиковую разницу сигнала и шума (PSNR) между a и b.

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

Предназначено для использования на сигналах (или изображениях). Производит значение 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.