Update predict.py

ultralytics · Feb 1, 2024 · 374c0d0 · 374c0d0
1 parent f4cd937
commit 374c0d0
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/ultralytics/models/yolo/segment/predict.py b/ultralytics/models/yolo/segment/predict.py
@@ -41,9 +41,7 @@ def postprocess(self, preds, img, orig_imgs):
             orig_imgs = ops.convert_torch2numpy_batch(orig_imgs)
 
         results = []
-        proto = (
-            preds[1][-1] if isinstance(preds[1], tuple) else preds[1]
-        )  # second output is tuple if pt, but an array if exported
+        proto = preds[1][-1] if isinstance(preds[1], tuple) else preds[1]  # second output is tuple if pt, but array if exported
         for i, pred in enumerate(p):
             orig_img = orig_imgs[i]
             img_path = self.batch[0][i]