fix

apperception-db · Aug 10, 2023 · 5af81b3 · 5af81b3
1 parent d10d591
commit 5af81b3
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/spatialyze/world.py b/spatialyze/world.py
@@ -11,6 +11,7 @@
 from .video_processor.payload import Payload
 from .video_processor.pipeline import Pipeline
 from .video_processor.stages.decode_frame.decode_frame import DecodeFrame
+from .video_processor.stages.decode_frame.parallel_decode_frame import ParallelDecodeFrame
 from .video_processor.stages.detection_2d.object_type_filter import ObjectTypeFilter
 from .video_processor.stages.detection_2d.yolo_detection import YoloDetection
 from .video_processor.stages.detection_3d.from_detection_2d_and_road import (
@@ -101,10 +102,10 @@ def _execute(world: "World"):
     objtypes_filter = ObjectTypeFilter(predicate=world.predicates)
     pipeline = Pipeline(
         [
-            DecodeFrame(),
+            ParallelDecodeFrame(),
             InView(distance=50, predicate=world.predicates),
             YoloDetection(),
-            objtypes_filter,
+            # objtypes_filter,
             FromDetection2DAndRoad(),
             *(
                 [DetectionEstimation()]

diff --git a/tests/workflow/test_simple_workflow.py b/tests/workflow/test_simple_workflow.py
@@ -1,4 +1,5 @@
 import pickle
+import json
 import os
 from os import environ
 import psycopg2
@@ -47,7 +48,7 @@ def test_simple_workflow():
     objects = _execute(world)
 
     with open(os.path.join(OUTPUT_DIR, f'simple-workflow.json'), 'w') as f:
-        f.write(objects, indent=1)
+        json.dump(objects, f, indent=1)
 
     with open(os.path.join(OUTPUT_DIR, f'simple-workflow.json'), 'r') as f:
         objects_groundtruth = f.read()