Fix dataset category indexes check (#18840)

2025-01-23 20:20:43 +08:00 · 2025-01-23 20:20:43 +08:00 · b30021e781
commit b30021e781
parent e170d50665
1 changed files with 1 additions and 1 deletions
--- a/ultralytics/data/utils.py
+++ b/ultralytics/data/utils.py
@ -136,7 +136,7 @@ def verify_image_label(args):

                # All labels
                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"Possible class labels are 0-{num_cls - 1}"
                )