Update ultralytics/yolo/engine/predictor.py

ultralytics/yolo/engine/predictor.py

@@ -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))
+                res = self.write_results(i, preds, (p, im, im0s))
+                s += res[0]
+                return res[1]
+                
 
                 if self.args.show:
                     self.show(p)