From 41f94d74aa62fc0e395ad4e581de805cf0de6bad Mon Sep 17 00:00:00 2001 From: Théo de la Hogue Date: Mon, 11 Mar 2024 09:21:21 +0100 Subject: Fixing transition matrix display. --- src/argaze/utils/demo_gaze_analysis_run.py | 16 +++++++++------- 1 file changed, 9 insertions(+), 7 deletions(-) diff --git a/src/argaze/utils/demo_gaze_analysis_run.py b/src/argaze/utils/demo_gaze_analysis_run.py index 5f58349..bd11ad5 100644 --- a/src/argaze/utils/demo_gaze_analysis_run.py +++ b/src/argaze/utils/demo_gaze_analysis_run.py @@ -100,16 +100,18 @@ def main(): # Iterate over columns (destinations) for to_aoi, probability in row.items(): - if from_aoi != to_aoi and probability > 0.0: + if from_aoi != GazeFeatures.OutsideAOI and to_aoi != GazeFeatures.OutsideAOI: - from_center = ar_frame.layers["demo_layer"].aoi_scene[from_aoi].center.astype(int) - to_center = ar_frame.layers["demo_layer"].aoi_scene[to_aoi].center.astype(int) - start_line = (0.5 * from_center + 0.5 * to_center).astype(int) + if from_aoi != to_aoi and probability > 0.0: - color = [int(probability*200) + 55, int(probability*200) + 55, int(probability*200) + 55] + from_center = ar_frame.layers["demo_layer"].aoi_scene[from_aoi].center.astype(int) + to_center = ar_frame.layers["demo_layer"].aoi_scene[to_aoi].center.astype(int) + start_line = (0.5 * from_center + 0.5 * to_center).astype(int) - cv2.line(frame_image, start_line, to_center, color, int(probability*10) + 2) - cv2.line(frame_image, from_center, to_center, [55, 55, 55], 2) + color = [int(probability*200) + 55, int(probability*200) + 55, int(probability*200) + 55] + + cv2.line(frame_image, start_line, to_center, color, int(probability*10) + 2) + cv2.line(frame_image, from_center, to_center, [55, 55, 55], 2) except KeyError: pass -- cgit v1.1