update type

apperception-db · Aug 14, 2023 · c25ee7e · c25ee7e
1 parent 30618f9
commit c25ee7e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/spatialyze/video_processor/stages/tracking_3d/from_tracking_2d_and_detection_3d.py b/spatialyze/video_processor/stages/tracking_3d/from_tracking_2d_and_detection_3d.py
@@ -48,8 +48,8 @@ def _run(self, payload: "Payload") -> "tuple[bitarray | None, dict[str, list] |
                     t.frame_idx,
                     t.detection_id,
                     t.object_id,
-                    pfc,
-                    p,
+                    tuple(pfc),
+                    tuple(p),
                     t.bbox_left,
                     t.bbox_top,
                     t.bbox_w,

diff --git a/spatialyze/video_processor/stages/tracking_3d/tracking_3d.py b/spatialyze/video_processor/stages/tracking_3d/tracking_3d.py
@@ -35,7 +35,7 @@ def encode_json(cls, o: "Any"):
                 "detection_id": tuple(o.detection_id),
                 "object_id": o.object_id,
                 "point_from_camera": o.point_from_camera,
-                "point": o.point.tolist(),
+                "point": o.point,
                 "bbox_left": o.bbox_left,
                 "bbox_top": o.bbox_top,
                 "bbox_w": o.bbox_w,