From 59ed28117275827303854cabe50339c34499f5aa Mon Sep 17 00:00:00 2001 From: Théo de la Hogue Date: Mon, 21 Nov 2022 19:34:29 +0100 Subject: Fixing type casting. --- src/argaze/utils/tobii_segment_aruco_aoi_export.py | 2 +- src/argaze/utils/tobii_stream_aruco_aoi_display.py | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) (limited to 'src') diff --git a/src/argaze/utils/tobii_segment_aruco_aoi_export.py b/src/argaze/utils/tobii_segment_aruco_aoi_export.py index 0e9e302..9f6ae78 100644 --- a/src/argaze/utils/tobii_segment_aruco_aoi_export.py +++ b/src/argaze/utils/tobii_segment_aruco_aoi_export.py @@ -259,7 +259,7 @@ def main(): # Merge all 2D aoi into a single 2D scene aoi2D_merged_scene = AOI2DScene.AOI2DScene() for name, aoi_array in aoi2D_dict.items(): - aoi2D_merged_scene[name] = numpy.sum(aoi_array, axis=0) / len(aoi_array) + aoi2D_merged_scene[name] = AOIFeatures.AreaOfInterest(numpy.sum(aoi_array, axis=0) / len(aoi_array)) aoi2D_merged_scene.draw(visu_frame.matrix, (0, 0)) diff --git a/src/argaze/utils/tobii_stream_aruco_aoi_display.py b/src/argaze/utils/tobii_stream_aruco_aoi_display.py index 15f1fde..c5becff 100644 --- a/src/argaze/utils/tobii_stream_aruco_aoi_display.py +++ b/src/argaze/utils/tobii_stream_aruco_aoi_display.py @@ -182,7 +182,7 @@ def main(): # When head is moving, ArUco tracking could return bad pose estimation and so bad AOI scene projection if head_moving: - ts_aois_scenes[round(video_ts_ms)] = AOIScene2D.AOI2DScene() + ts_aois_scenes[round(video_ts_ms)] = AOI2DScene.AOI2DScene() raise UserWarning('Head is moving') @@ -197,7 +197,7 @@ def main(): # When no marker is detected, no AOI scene projection can't be done if aruco_tracker.tracked_markers_number == 0: - ts_aois_scenes[round(video_ts_ms)] = AOIScene2D.AOI2DScene() + ts_aois_scenes[round(video_ts_ms)] = AOI2DScene.AOI2DScene() raise UserWarning('No marker detected') @@ -240,7 +240,7 @@ def main(): # Merge all 2D aoi into a single 2D scene aoi2D_merged_scene = AOI2DScene.AOI2DScene() for name, aoi_array in aoi2D_dict.items(): - aoi2D_merged_scene[name] = numpy.sum(aoi_array, axis=0) / len(aoi_array) + aoi2D_merged_scene[name] = AOIFeatures.AreaOfInterest(numpy.sum(aoi_array, axis=0) / len(aoi_array)) aoi2D_merged_scene.draw(visu_frame.matrix) -- cgit v1.1