def compute_visuals(self): self.textures_unnormalized = unnormalize(self.textures, *self.opt.texture_norm_stats) self.textures_unnormalized = draw_rois_on_texture( self.rois, self.textures_unnormalized) self.cloths_decoded = decode_cloth_labels(self.cloths) self.fakes_scaled = scale_tensor(self.fakes, scale_each=True) if self.is_train: self.targets_unnormalized = unnormalize( self.targets, *self.opt.texture_norm_stats)
def compute_visuals(self): self.cloth_decoded = decode_cloth_labels(self.real_A) self.fakes_scaled = scale_tensor(self.fake_B) self.textures_unnormalized = unnormalize(self.real_B, *self.opt.texture_norm_stats)