fix: add images with no annotations to LuxonisDataset

luxonis · Nov 1, 2024 · f763786 · f763786
1 parent 749696d
commit f763786
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/datadreamer/utils/luxonis_dataset_converter.py b/datadreamer/utils/luxonis_dataset_converter.py
@@ -80,6 +80,12 @@ def dataset_generator():
                 image_full_path = os.path.join(dataset_dir, image_path)
                 width, height = Image.open(image_full_path).size
                 labels = data[image_path]["labels"]
+
+                if len(labels) == 0:
+                    yield {
+                        "file": image_full_path,
+                    }
+
                 for label in labels:
                     yield {
                         "file": image_full_path,