Rename predictor postprocess orig_img to orig_imgs (#2877)

Signed-off-by: Afif Al Mamun <afifanik@gmail.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-05-28 21:54:05 +06:00
parent dfe7f627d8
commit 75012e575e
2 changed files with 3 additions and 3 deletions
--- a/ultralytics/yolo/engine/predictor.py
+++ b/ultralytics/yolo/engine/predictor.py
@ -171,7 +171,7 @@ class BasePredictor:

        return log_string

-    def postprocess(self, preds, img, orig_img):
+    def postprocess(self, preds, img, orig_imgs):
        """Post-processes predictions for an image and returns them."""
        return preds