def postprocess(self, corrupted, masks, generated): corrupted = corrupted * 0.5 + 0.5 generated = generated * 0.5 + 0.5 corrupted = corrupted.permute(0, 3, 2, 1).cpu().numpy() processed = generated.permute(0, 3, 2, 1).cpu().detach().numpy() masks = np.transpose(masks, axes=(0, 3, 2, 1)) for i in range(len(processed)): processed[i] = blending.blend(corrupted[i], processed[i], 1 - masks[i]) processed = torch.tensor(processed).permute(0, 3, 2, 1) return (processed * 2.0 - 1.0).cuda()
def poissonblending(img1, img2, mask): """Helper: interface to external poisson blending""" return blending.blend(img1, img2, 1 - mask)
def poissonblending(img1, img2, mask): """Helper: interface to external poisson blending""" return blending.blend(img1, img2, 1 - mask)