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