fixed bug of whitespace (#1173)

Deci-AI · Jun 14, 2023 · d95fd0b · d95fd0b
1 parent 7e45a68
commit d95fd0b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/super_gradients/training/datasets/detection_datasets/yolo_format_detection.py b/src/super_gradients/training/datasets/detection_datasets/yolo_format_detection.py
@@ -228,7 +228,7 @@ def _parse_yolo_label_file(label_file_path: str, ignore_invalid_labels: bool = T
         labels_yolo_format, invalid_labels = [], []
         for line in filter(lambda x: x != "\n", lines):
             try:
-                label_id, cx, cw, w, h = line.split(" ")
+                label_id, cx, cw, w, h = line.split()
                 labels_yolo_format.append([int(label_id), float(cx), float(cw), float(w), float(h)])
             except Exception as e:
                 if ignore_invalid_labels: