diff options
| -rw-r--r-- | src/argaze/utils/tobii_segment_aruco_aoi_export.py | 2 | ||||
| -rw-r--r-- | src/argaze/utils/tobii_stream_aruco_aoi_display.py | 6 |
2 files changed, 4 insertions, 4 deletions
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) |