Update predict.py

predict.py

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