Function compute_psnr

pointersect/inference/inference_utils.py:194–215

Compute peak signal to noise ratio Args: arr: (*b_shape, *d_shape) ref: (*b_shape, *d_shape) ndim_b: number of dimension of b_shape. If None, = 0. Returns: psnr: (*b_shape,)

(
        arr: torch.Tensor,
        ref: torch.Tensor,
        ndim_b: int = None,
        max_val: float = 1.,
        valid_mask: torch.Tensor = None,
)

def compute_psnr(
        arr: torch.Tensor,
        ref: torch.Tensor,
        ndim_b: int = None,
        max_val: float = 1.,
        valid_mask: torch.Tensor = None,
):
    """
    Compute peak signal to noise ratio
    Args:
        arr: (*b_shape, *d_shape)
        ref: (*b_shape, *d_shape)
        ndim_b:
            number of dimension of b_shape. If None, = 0.

    Returns:
        psnr: (*b_shape,)
    """

    mse = compute_mse(arr=arr, ref=ref, ndim_b=ndim_b, valid_mask=valid_mask)  # (*b,)
    psnr = 10 * torch.log10((max_val * max_val) / mse)  # (*b,)
    return psnr


def compute_ssim(