Replace enumerate with zip in models/yolo (#14599)
This commit is contained in:
parent
22a44d82c5
commit
db82d1c6ae
4 changed files with 4 additions and 12 deletions
|
|
@ -42,9 +42,7 @@ class SegmentationPredictor(DetectionPredictor):
|
|||
|
||||
results = []
|
||||
proto = preds[1][-1] if isinstance(preds[1], tuple) else preds[1] # tuple if PyTorch model or array if exported
|
||||
for i, pred in enumerate(p):
|
||||
orig_img = orig_imgs[i]
|
||||
img_path = self.batch[0][i]
|
||||
for i, (pred, orig_img, img_path) in enumerate(zip(p, orig_imgs, self.batch[0])):
|
||||
if not len(pred): # save empty boxes
|
||||
masks = None
|
||||
elif self.args.retina_masks:
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue