ultralytics 8.0.153 YOLO Tasks Cleanup (#4314)

2023-08-12 02:30:57 +02:00
parent 39395aedc8
commit 822608986c
22 changed files with 87 additions and 55 deletions
--- a/ultralytics/models/sam/predict.py
+++ b/ultralytics/models/sam/predict.py
@ -310,7 +310,7 @@ class Predictor(BasePredictor):
        self.done_warmup = True

    def postprocess(self, preds, img, orig_imgs):
-        """Postprocesses inference output predictions to create detection masks for objects."""
+        """Post-processes inference output predictions to create detection masks for objects."""
        # (N, 1, H, W), (N, 1)
        pred_masks, pred_scores = preds[:2]
        pred_bboxes = preds[2] if self.segment_all else None