Update predict.py

predict.py

@@ -53,7 +53,7 @@ def draw_semantic_segmentation(segmentation_map, image, palette):
 
 def visualize_instance_seg_mask(mask, input_image):
     color_segmentation_map = np.zeros((mask.shape[0], mask.shape[1], 3), dtype=np.uint8)
-    print("color_segmentation_map:",color_segmentation_map)
+    print("segmentation_map:",mask)
     labels = np.unique(mask)
     print("labels:",labels)
     label2color = {int(label): (random.randint(0, 1), random.randint(0, 255), random.randint(0, 255)) for label in labels}