diff --git a/spatialyze/world.py b/spatialyze/world.py index 536cfff..95aa9c5 100644 --- 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 index eb1d98e..22d852f 100644 --- 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()