Spaces:

kisa-misa
/

YOLOv8-real-time

Build error

kisa-misa commited on May 23, 2023

Commit

7da7d10

1 Parent(s): 4d7941e

Update ultralytics/yolo/engine/predictor.py

Files changed (1) hide show

ultralytics/yolo/engine/predictor.py CHANGED Viewed

@@ -186,7 +186,10 @@ class BasePredictor:
                 if self.webcam:
                     path, im0s = path[i], im0s[i]
                 p = Path(path)
-                s += self.write_results(i, preds, (p, im, im0s))
                 if self.args.show:
                     self.show(p)

                 if self.webcam:
                     path, im0s = path[i], im0s[i]
                 p = Path(path)
+                res = self.write_results(i, preds, (p, im, im0s))
+                s += res[0]
+                return res[1]
                 if self.args.show:
                     self.show(p)