diff options
| author | Théo de la Hogue | 2024-03-11 09:21:21 +0100 |
|---|---|---|
| committer | Théo de la Hogue | 2024-03-11 09:21:21 +0100 |
| commit | 41f94d74aa62fc0e395ad4e581de805cf0de6bad (patch) | |
| tree | 17fbdde7157692ace2fbd817922989925bd0c09a | |
| parent | 2f4b9934a5cb60b43bb5f378056c10a39aec279d (diff) | |
| download | argaze-41f94d74aa62fc0e395ad4e581de805cf0de6bad.zip argaze-41f94d74aa62fc0e395ad4e581de805cf0de6bad.tar.gz argaze-41f94d74aa62fc0e395ad4e581de805cf0de6bad.tar.bz2 argaze-41f94d74aa62fc0e395ad4e581de805cf0de6bad.tar.xz | |
Fixing transition matrix display.
| -rw-r--r-- | src/argaze/utils/demo_gaze_analysis_run.py | 16 |
1 files 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 |