From c270539e24c3f1b2ad56b259acc81e50cee30124 Mon Sep 17 00:00:00 2001 From: ofri masad Date: Wed, 14 Jun 2023 12:35:52 +0300 Subject: [PATCH] fixed bug of whitespace --- .../datasets/detection_datasets/yolo_format_detection.py | 2 +- 1 file changed, 1 insertion(+), 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 index 804990f1ff..999f48dcfb 100644 --- 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: