def _compute_acc(self):
target_foreground = self.batch['B'].to(device).detach()
target_alpha_map = self.batch['ALPHA'].to(device).detach()
foreground = self.G_pred_foreground.detach()
alpha_map = self.G_pred_alpha.detach()
psnr1 = utils.cpt_batch_psnr(foreground,
target_foreground,
PIXEL_MAX=1.0)
psnr2 = utils.cpt_batch_psnr(alpha_map,
target_alpha_map,
PIXEL_MAX=1.0)
return (psnr1 + psnr2) / 2.0
def _compute_acc(self):
"""Accuracy is here the PSNR: https://en.wikipedia.org/wiki/Peak_signal-to-noise_ratio
"""
target_foreground = self.batch['B'].to(device).detach()
target_alpha_map = self.batch['ALPHA'].to(device).detach()
foreground = self.G_pred_foreground.detach()
alpha_map = self.G_pred_alpha.detach()
psnr1 = utils.cpt_batch_psnr(foreground,
target_foreground,
PIXEL_MAX=1.0)
psnr2 = utils.cpt_batch_psnr(alpha_map,
target_alpha_map,
PIXEL_MAX=1.0)
return (psnr1 + psnr2) / 2.0
def _compute_acc(self):
target = self.img_batch.detach()
canvas = self.G_pred_canvas.detach()
psnr = utils.cpt_batch_psnr(canvas, target, PIXEL_MAX=1.0)
return psnr
def _compute_acc(self):
target = self.batch['B'].to(device).detach()
G_pred = self.G_pred.detach()
acc = utils.cpt_batch_psnr(G_pred, target, PIXEL_MAX=1.0)
return acc
def _compute_acc(self):
target = self.batch['dmap'].to(device).detach()
img = self.G_pred.detach()
psnr = utils.cpt_batch_psnr(img, target, PIXEL_MAX=1.0)
return psnr
