Merge branch 'ck/benchmark' of github.com:apperception-db/spatialyze …

…into ck/benchmark
apperception-db · Aug 11, 2023 · a04ab08 · a04ab08
2 parents c9251de + f61079b
commit a04ab08
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
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
@@ -17,7 +17,9 @@ def _run(self, payload: "Payload") -> "tuple[bitarray | None, dict[str, list] |
         detections = Detection3D.get(payload.metadata)
         assert detections is not None
 
-        for k, detection, tracking, frame in zip(payload.keep, detections, trackings, payload.video):
+        for k, detection, tracking, frame in zip(
+            payload.keep, detections, trackings, payload.video
+        ):
             dets, _, dids = detection
             if not k or tracking is None or detection is None:
                 metadata.append(dict())

diff --git a/spatialyze/video_processor/utils/insert_trajectory.py b/spatialyze/video_processor/utils/insert_trajectory.py
@@ -1,4 +1,5 @@
 import datetime
+
 import numpy as np
 import numpy.typing as npt