From a561a6de8e479c3ff52030577da4b5cd54fc6e10 Mon Sep 17 00:00:00 2001 From: Martin Kozlovsky Date: Tue, 3 Sep 2024 06:46:19 +0200 Subject: [PATCH] fixed loading annotations --- luxonis_ml/data/loaders/luxonis_loader.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/luxonis_ml/data/loaders/luxonis_loader.py b/luxonis_ml/data/loaders/luxonis_loader.py index 596a8e55..a5ad2c27 100644 --- a/luxonis_ml/data/loaders/luxonis_loader.py +++ b/luxonis_ml/data/loaders/luxonis_loader.py @@ -234,13 +234,13 @@ def _load_image_with_annotations(self, idx: int) -> Tuple[np.ndarray, Labels]: data = json.loads(ann_str) if type_ == "ArrayAnnotation" and self.dataset.is_remote: data["path"] = self.dataset.arrays_path / data["path"] - data.update( - { - "class": class_, - "task": task, - "instance_id": instance_id, - } - ) + data.update( + { + "class": class_, + "task": task, + "instance_id": instance_id, + } + ) annotation = load_annotation(type_, data) labels_by_task[task].append(annotation)