ultralytics 8.0.153 YOLO Tasks Cleanup (#4314)

2023-08-12 02:30:57 +02:00 · 2023-08-12 02:30:57 +02:00 · 822608986c
commit 822608986c
parent 39395aedc8
22 changed files with 87 additions and 55 deletions
--- a/ultralytics/models/yolo/classify/predict.py
+++ b/ultralytics/models/yolo/classify/predict.py
@ -21,7 +21,7 @@ class ClassificationPredictor(BasePredictor):
        return img.half() if self.model.fp16 else img.float()  # uint8 to fp16/32

    def postprocess(self, preds, img, orig_imgs):
-        """Postprocesses predictions to return Results objects."""
+        """Post-processes predictions to return Results objects."""
        results = []
        for i, pred in enumerate(preds):
            orig_img = orig_imgs[i] if isinstance(orig_imgs, list) else orig_imgs