Replace enumerate with zip in models/yolo (#14599)

2024-07-23 01:52:43 +08:00 · 2024-07-23 01:52:43 +08:00 · db82d1c6ae
commit db82d1c6ae
parent 22a44d82c5
4 changed files with 4 additions and 12 deletions
--- a/ultralytics/models/yolo/detect/predict.py
+++ b/ultralytics/models/yolo/detect/predict.py
@ -35,9 +35,7 @@ class DetectionPredictor(BasePredictor):
            orig_imgs = ops.convert_torch2numpy_batch(orig_imgs)

        results = []
-        for i, pred in enumerate(preds):
-            orig_img = orig_imgs[i]
+        for pred, orig_img, img_path in zip(preds, orig_imgs, self.batch[0]):
            pred[:, :4] = ops.scale_boxes(img.shape[2:], pred[:, :4], orig_img.shape)
-            img_path = self.batch[0][i]
            results.append(Results(orig_img, path=img_path, names=self.model.names, boxes=pred))
        return results