tfmri.metrics.PSNR

class PSNR(*args, **kwargs)[source]

Bases: tensorflow_mri.python.metrics.iqa_metrics.MeanMetricWrapperIQA

Peak signal-to-noise ratio (PSNR) metric.

The PSNR is the ratio between the maximum possible power of an image and the power of corrupting noise, estimated by comparing to a reference image.

This metric supports 2D and 3D image inputs, y_true and y_pred. For 2D images, inputs must have rank >= 3 with shape batch_shape + [height, width, channels]. For 3D images, inputs must have rank >= 4 with shape batch_shape + [depth, height, width, channels]. If multichannel is False, the channel dimension should be omitted.

Parameters

  • max_val – The dynamic range of the images (i.e., the difference between the maximum and the minimum allowed values). Defaults to 1 for floating point input images and MAX for integer input images, where MAX is the largest positive representable number for the data type.

  • batch_dims –

    An int. The number of batch dimensions in input images. If None, it is inferred from inputs and image_dims as (rank of inputs) - image_dims - 1. If image_dims is also None, then batch_dims defaults to 1. batch_dims can always be inferred if image_dims was specified, so you only need to provide one of the two.

  • image_dims –

    An int. The number of spatial dimensions in input images. If None, it is inferred from inputs and batch_dims as (rank of inputs) - batch_dims - 1. Defaults to None. image_dims can always be inferred if batch_dims was specified, so you only need to provide one of the two.

  • rank –

    An int. The number of spatial dimensions. Must be 2 or 3. Defaults to tf.rank(y_true) - 2. In other words, if rank is not explicitly set, y_true and y_pred should have shape [batch, height, width, channels] if processing 2D images or [batch, depth, height, width, channels] if processing 3D images.

  • multichannel –

    A boolean. Whether multichannel computation is enabled. If False, the inputs y_true and y_pred are not expected to have a channel dimension, i.e. they should have shape batch_shape + [height, width] (2D) or batch_shape + [depth, height, width] (3D).

  • complex_part – The part of a complex input to be used in the computation of the metric. Must be one of 'real', 'imag', 'abs' or 'angle'. Note that real and imaginary parts, as well as angles, will be scaled to avoid negative numbers.

  • name – String name of the metric instance.

  • dtype – Data type of the metric result.

DEPRECATED FUNCTION ARGUMENTS

Deprecated: SOME ARGUMENTS ARE DEPRECATED: (rank). They will be removed after 2022-09-01. Instructions for updating: Use argument image_dims instead.