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