fix native parser file paths

luxonis · Jan 14, 2025 · 590fda8 · 590fda8
1 parent 8de4ad0
commit 590fda8
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/luxonis_ml/data/parsers/native_parser.py b/luxonis_ml/data/parsers/native_parser.py
@@ -66,7 +66,13 @@ def from_split(self, annotation_path: Path) -> ParserOutput:
         """
 
         def generator() -> DatasetIterator:
-            yield from json.loads(annotation_path.read_text())
+            for record in json.loads(annotation_path.read_text()):
+                if "file" not in record:
+                    raise ValueError(
+                        "The annotation record must contain 'file' key."
+                    )
+                record["file"] = annotation_path.parent / record["file"]
+                yield record
 
         added_images = self._get_added_images(generator())