aboutsummaryrefslogtreecommitdiff
path: root/src/argaze/utils/export_tobii_segment_aruco_visual_scan.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/argaze/utils/export_tobii_segment_aruco_visual_scan.py')
-rw-r--r--src/argaze/utils/export_tobii_segment_aruco_visual_scan.py7
1 files changed, 5 insertions, 2 deletions
diff --git a/src/argaze/utils/export_tobii_segment_aruco_visual_scan.py b/src/argaze/utils/export_tobii_segment_aruco_visual_scan.py
index 25babf6..6b0a41f 100644
--- a/src/argaze/utils/export_tobii_segment_aruco_visual_scan.py
+++ b/src/argaze/utils/export_tobii_segment_aruco_visual_scan.py
@@ -212,7 +212,7 @@ def main():
continue
# Track markers with pose estimation and draw them
- aruco_tracker.track(video_frame.matrix)
+ aruco_tracker.track(video_frame.matrix, check_rotation=False)
aruco_tracker.draw(video_frame.matrix)
# Draw focus area
@@ -312,11 +312,14 @@ def main():
# Print aruco tracking metrics
print('\nAruco marker tracking metrics')
- try_count, tracked_counts = aruco_tracker.get_track_metrics()
+ try_count, tracked_counts, rejected_counts = aruco_tracker.get_track_metrics()
for marker_id, tracked_count in tracked_counts.items():
print(f'Markers {marker_id} has been detected in {tracked_count} / {try_count} frames ({round(100 * tracked_count / try_count, 2)} %)')
+ for marker_id, rejected_count in rejected_counts.items():
+ print(f'Markers {marker_id} has been rejected in {rejected_count} / {try_count} frames ({round(100 * rejected_count / try_count, 2)} %)')
+
# Build visual scan based on a pointer position
visual_scan = GazeFeatures.PointerBasedVisualScan(ts_aois_scenes, ts_gaze_positions)
print(f'{len(visual_scan.steps())} visual scan steps found')