Update visualization.py

visualization.py

@@ -11,10 +11,6 @@ from utils import fig2img
 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()