def calc_metrics(im_pred, im_gt, mask=None):
q_im_pred = quantize(im_pred.data, rgb_range=1.)
q_im_gt = quantize(im_gt.data, rgb_range=1.)
if mask is not None:
q_im_pred = q_im_pred * mask
q_im_gt = q_im_gt * mask
psnr = calc_psnr(q_im_pred, q_im_gt, mask=mask)
# ssim = calc_ssim(q_im_pred.cpu(), q_im_gt.cpu()) # This function using SciPy compare_ssim() is very, very slow
ssim = ssim_pth(q_im_pred.unsqueeze(0), q_im_gt.unsqueeze(0), val_range=255)
return psnr, ssim
def calc_metrics(im_pred, im_gt, mask=None):
q_im_pred = quantize(im_pred.data, rgb_range=1.)
q_im_gt = quantize(im_gt.data, rgb_range=1.)
if mask is not None:
q_im_pred = q_im_pred * mask
q_im_gt = q_im_gt * mask
psnr = calc_psnr(q_im_pred, q_im_gt, mask=mask)
# ssim = calc_ssim(q_im_pred.cpu(), q_im_gt.cpu())
ssim = ssim_pth(q_im_pred.unsqueeze(0),
q_im_gt.unsqueeze(0),
val_range=255)
return psnr, ssim