diff --git a/apps/data-collection/backend/src/core/snapping/snaps_producer.py b/apps/data-collection/backend/src/core/snapping/snaps_producer.py index 70e5be3f3..c44ca4dbe 100644 --- a/apps/data-collection/backend/src/core/snapping/snaps_producer.py +++ b/apps/data-collection/backend/src/core/snapping/snaps_producer.py @@ -45,12 +45,12 @@ def process( detections=detections.detections, tracklets=tracklets ): continue - + file_name = f"{cond.name}_{int(time.time())}" + file_group = dai.FileGroup() + file_group.addImageDetectionsPair(file_name, frame, detections) snap = SnapData( snap_name=cond.name, - file_name=f"{cond.name}_{int(time.time())}", - frame=frame, - detections=detections, + file_group=file_group, tags=cond.tags, extras=cond.make_extras(), ) diff --git a/apps/data-collection/backend/src/requirements.txt b/apps/data-collection/backend/src/requirements.txt index ac38434af..ad50c7cf2 100644 --- a/apps/data-collection/backend/src/requirements.txt +++ b/apps/data-collection/backend/src/requirements.txt @@ -1,5 +1,5 @@ depthai==3.2.1 -depthai-nodes==0.3.7 +depthai-nodes @ git+https://github.com/luxonis/depthai-nodes.git@feat/dai_dev_req opencv-python-headless~=4.10.0 numpy>=1.22 tokenizers~=0.21.0 diff --git a/integrations/hub-snaps-events/requirements.txt b/integrations/hub-snaps-events/requirements.txt index bd5223a56..0a1a51659 100644 --- a/integrations/hub-snaps-events/requirements.txt +++ b/integrations/hub-snaps-events/requirements.txt @@ -1,4 +1,5 @@ depthai==3.2.1 -depthai-nodes==0.3.7 +#depthai-nodes==0.3.7 +depthai-nodes @ git+https://github.com/luxonis/depthai-nodes.git@feat/dai_dev_req python-dotenv diff --git a/integrations/hub-snaps-events/utils/snaps_producer.py b/integrations/hub-snaps-events/utils/snaps_producer.py index 00329d444..36e17f296 100644 --- a/integrations/hub-snaps-events/utils/snaps_producer.py +++ b/integrations/hub-snaps-events/utils/snaps_producer.py @@ -42,11 +42,11 @@ def process( return if time.time() - self._last_sent >= self._time_interval: + file_group = dai.FileGroup() + file_group.addImageDetectionsPair(None, frame, detections) snap = SnapData( snap_name="test_snap", - file_name=None, - frame=frame, - detections=detections, + file_group=file_group, tags=["test_tag"], extras={"extra_key": "extra_value"}, )