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

…into ck/benchmark
apperception-db · Oct 25, 2023 · d1a7738 · d1a7738
2 parents 9d88ffc + 2eff9e6
commit d1a7738
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 9 deletions.
diff --git a/spatialyze/video_processor/stages/detection_estimation/__init__.py b/spatialyze/video_processor/stages/detection_estimation/__init__.py
@@ -78,17 +78,17 @@ def _run(self, payload: "Payload"):
             if i != next_frame_num:
                 skipped_frame_num.append(i)
                 metadata.append([])
-                print('skip   ', i, next_frame_num)
+                print("skip   ", i, next_frame_num)
                 continue
 
             next_frame_num = i + 1
 
             det, _, dids = dets[i]
-            print('--new    ', i, new_car(dets, i, i + 5))
+            print("--new    ", i, new_car(dets, i, i + 5))
             if new_car(dets, i, i + 5) <= i + 1:
                 # will not map segment if cannot skip in the first place
                 metadata.append([])
-                print('new    ', i, new_car(dets, i, i + 5))
+                print("new    ", i, new_car(dets, i, i + 5))
                 continue
 
             start_detection_time = time.time()
@@ -108,7 +108,7 @@ def _run(self, payload: "Payload"):
             if len(det) == 0 or len(all_detection_info_pruned) == 0:
                 # skipped_frame_num.append(i)
                 metadata.append([])
-                print('0      ', i)
+                print("0      ", i)
                 continue
 
             start_generate_sample_plan = time.time()
@@ -119,7 +119,7 @@ def _run(self, payload: "Payload"):
             next_frame_num = next_sample_plan.get_next_frame_num()
             next_frame_num = new_car(dets, i, next_frame_num)
             logger.info(f"founded next_frame_num {next_frame_num}")
-            print('next   ', i)
+            print("next   ", i)
             metadata.append(all_detection_info)
 
             next_action_type = next_sample_plan.get_action_type()

diff --git a/spatialyze/video_processor/stream/exit_frame_sampler.py b/spatialyze/video_processor/stream/exit_frame_sampler.py
@@ -3,9 +3,6 @@
 from collections.abc import Iterable, Iterator
 from typing import Generic, TypeVar
 
-import shapely
-import shapely.wkb
-
 from ..stages.detection_estimation import (
     construct_estimated_all_detection_info,
     generate_sample_plan_once,
@@ -14,7 +11,7 @@
     trajectory_3d,
 )
 from ..video import Video
-from .data_types import Detection3D, Skip, skip
+from .data_types import Detection3D, Skip
 from .stream import Stream
 
 logging.basicConfig()