diff --git a/utils/dataloaders.py b/utils/dataloaders.py index 675c2898e7d7..f027307ccb94 100755 --- a/utils/dataloaders.py +++ b/utils/dataloaders.py @@ -963,7 +963,7 @@ def verify_image_label(args): if len(i) < nl: # duplicate row check lb = lb[i] # remove duplicates if segments: - segments = segments[i] + segments = [segments[x] for x in i] msg = f'{prefix}WARNING: {im_file}: {nl - len(i)} duplicate labels removed' else: ne = 1 # label empty