Function psnr

script/models/metrics.py:12–13

(image_pred, image_gt, valid_mask=None, reduction='mean')

    return value

def psnr(image_pred, image_gt, valid_mask=None, reduction='mean'):
    return -10*torch.log10(mse(image_pred, image_gt, valid_mask, reduction))

def ssim(image_pred, image_gt, reduction='mean'):
    """