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/nas/predict.py
+++ b/ultralytics/models/nas/predict.py
@ -11,7 +11,7 @@ from ultralytics.utils.ops import xyxy2xywh
 class NASPredictor(BasePredictor):

    def postprocess(self, preds_in, img, orig_imgs):
-        """Postprocesses predictions and returns a list of Results objects."""
+        """Postprocess predictions and returns a list of Results objects."""

        # Cat boxes and class scores
        boxes = xyxy2xywh(preds_in[0][0])