def outputs(self): _outputs = {"generated": self._generated} _outputs.update({"view0_mask00_rgb": nn.mask2rgb(self.m0)}) _outputs.update({"m0_sample": self.m0_sample}) _outputs.update({"out_parts_hard": self.out_parts_hard}) _outputs.update({"out_parts_soft": self.out_parts_soft}) # _outputs.update({"part_generated": self._part_generated}) return _outputs
def outputs(self): _outputs = {"generated": self._generated} _outputs.update({"view0_mask00_rgb": nn.mask2rgb(self.m0)}) _outputs.update({"m0_sample": self.m0_sample}) _outputs.update({"out_parts_hard": self.out_parts_hard}) _outputs.update({"out_parts_soft": self.out_parts_soft}) if self.use_tps: _outputs.update({ "tps_view0": self.augmented_views[0], "tps_view1": self.augmented_views[1], }) return _outputs
def visualize_mask(name, mask, img_ops, make_hot=True): n_parts = mask.shape.as_list()[3] vis_mask = nn.mask2rgb(mask, make_hot) img_ops[name + "_visualization"] = vis_mask