Duplicate segment verification fix (#9225)

Solution by @Laughing-q to resolve duplicate segment verification bug in #9052 (comment) Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
ultralytics · Aug 30, 2022 · 5fb267f · 5fb267f
1 parent f37ac8d
commit 5fb267f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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