Update predict.py

predict.py

@@ -53,8 +53,9 @@ 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)
     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}
 
     for label, color in label2color.items():