Update inference_seg.py

inference_seg.py

@@ -26,7 +26,6 @@ def load_model(use_box=False):
 def run(model, img_path, box=None, device="cpu"):
     output = model(img_path, box=box)
     mask = output
-    mask = (mask > 0).astype(np.uint8)
     return mask
 # import os
 # import torch