mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00
Code Issues Projects Releases Wiki Activity

np.unique() sort fix for segments (#5609)

Browse Source 
* `np.unique()` sort fix for segments

* Update datasets.py
This commit is contained in:
Glenn Jocher 2021-11-11 00:15:17 +01:00 committed by GitHub
parent 30bc089cbb
commit 69032519bc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
utils/datasets.py
View File

@ -913,10 +913,12 @@ def verify_image_label(args):
assert l.shape[1] == 5, f'labels require 5 columns, {l.shape[1]} columns detected' assert l.shape[1] == 5, f'labels require 5 columns, {l.shape[1]} columns detected'
assert (l >= 0).all(), f'negative label values {l[l < 0]}' assert (l >= 0).all(), f'negative label values {l[l < 0]}'
assert (l[:, 1:] <= 1).all(), f'non-normalized or out of bounds coordinates {l[:, 1:][l[:, 1:] > 1]}' assert (l[:, 1:] <= 1).all(), f'non-normalized or out of bounds coordinates {l[:, 1:][l[:, 1:] > 1]}'
l = np.unique(l, axis=0) # remove duplicate rows _, i = np.unique(l, axis=0, return_index=True)
if len(l) < nl: if len(i) < nl: # duplicate row check
segments = np.unique(segments, axis=0) l = l[i] # remove duplicates
msg = f'{prefix}WARNING: {im_file}: {nl - len(l)} duplicate labels removed' if segments:
segments = segments[i]
msg = f'{prefix}WARNING: {im_file}: {nl - len(i)} duplicate labels removed'
else: else:
ne = 1 # label empty ne = 1 # label empty
l = np.zeros((0, 5), dtype=np.float32) l = np.zeros((0, 5), dtype=np.float32)