Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

Duplicate segment verification fix (#9225)

Browse Source 
Solution by @Laughing-q  to resolve duplicate segment verification bug in https://github.com/ultralytics/yolov5/pull/9052#issuecomment-1231426638

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Glenn Jocher authored and GitHub committed 2022-08-30 13:40:51 +02:00
1 parent f37ac8d611
commit 5fb267f3e5
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
utils/dataloaders.py
+1 -1
View File
@@ -963,7 +963,7 @@ def verify_image_label(args):
if len(i) < nl: # duplicate row check if len(i) < nl: # duplicate row check
lb = lb[i] # remove duplicates lb = lb[i] # remove duplicates
if segments: if segments:
segments = segments[i] segments = [segments[x] for x in i]
msg = f'{prefix}WARNING: {im_file}: {nl - len(i)} duplicate labels removed' msg = f'{prefix}WARNING: {im_file}: {nl - len(i)} duplicate labels removed'
else: else:
ne = 1 # label empty ne = 1 # label empty