support visualizing panoptic segmentation prediction

Summary:
Pull Request resolved: https://github.com/facebookresearch/d2go/pull/532

Enable the visualization of panoptic segmentation.

Reviewed By: tglik

Differential Revision: D45334039

fbshipit-source-id: eebd9316d56d8132a5d3c166058ae18a0e88e928

d2go/utils/visualization.py

@@ -101,8 +101,10 @@ class VisualizerWrapper(object):
         visualizer = self.visualizer(image, metadata=MetadataCatalog.get(dataset_name))
 
         if "panoptic_seg" in output_dict:
-            # NOTE: refer to https://fburl.com/diffusion/evarrhbh
-            raise NotImplementedError()
+            panoptic_seg, segments_info = output_dict["panoptic_seg"]
+            visualizer.draw_panoptic_seg_predictions(
+                panoptic_seg.to("cpu"), segments_info
+            )
         if "instances" in output_dict:
             visualizer.draw_instance_predictions(output_dict["instances"].to("cpu"))
         if "sem_seg" in output_dict: