HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Fix dataset category indexes check (#18840)

Browse source
This commit is contained in:
Laughing 2025-01-23 20:20:43 +08:00 • committed by GitHub
parent e170d50665
commit b30021e781
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/data/utils.py
View file

@ -136,7 +136,7 @@ def verify_image_label(args):
# All labels # All labels
max_cls = lb[:, 0].max() # max label count max_cls = lb[:, 0].max() # max label count
assert max_cls <= num_cls, ( assert max_cls < num_cls, (
f"Label class {int(max_cls)} exceeds dataset class count {num_cls}. " f"Label class {int(max_cls)} exceeds dataset class count {num_cls}. "
f"Possible class labels are 0-{num_cls - 1}" f"Possible class labels are 0-{num_cls - 1}"
) )