Update visualization.py

visualization.py

@@ -19,6 +19,10 @@ def visualize_mask(mask, img, alpha=0.5):
 def visualize_prediction(
     pil_img, output_dict, threshold=0.7, id2label=None, display_mask=False, mask=None
 ):
+    print(display_mask)
+    print(mask)
+    print(type(mask))
+    
     keep = output_dict["scores"] > threshold
     boxes = output_dict["boxes"][keep].tolist()
     scores = output_dict["scores"][keep].tolist()