diff options
| -rw-r--r-- | src/argaze/utils/tobii_segment_display.py | 19 |
1 files changed, 7 insertions, 12 deletions
diff --git a/src/argaze/utils/tobii_segment_display.py b/src/argaze/utils/tobii_segment_display.py index a106e7b..604ebcf 100644 --- a/src/argaze/utils/tobii_segment_display.py +++ b/src/argaze/utils/tobii_segment_display.py @@ -94,9 +94,6 @@ def main(): gaze_position_pixel = GazeFeatures.GazePosition( (int(nearest_gaze_position.value[0] * video_frame.width), int(nearest_gaze_position.value[1] * video_frame.height)) ) - # Draw gaze position - cv.circle(video_frame.matrix, gaze_position_pixel, 2, (0, 255, 255), -1) - # Get nearest gaze position 3D before video timestamp and remove all gaze positions before _, nearest_gaze_position_3d = tobii_ts_gaze_positions_3d.pop_first_until(video_ts) @@ -108,9 +105,9 @@ def main(): gaze_position_pixel.accuracy = round(video_frame.width * float(gaze_accuracy_mm) / float(tobii_camera_hfov_mm)) - # Draw gaze accuracy - cv.circle(video_frame.matrix, gaze_position_pixel, gaze_position_pixel.accuracy, (0, 255, 255), 1) - + # Draw gaze + gaze_position_pixel.draw(video_frame.matrix) + # Wait for gaze position except ValueError: pass @@ -138,13 +135,11 @@ def main(): cv.rectangle(video_frame.matrix, (0, 0), (550, 50), (63, 63, 63), -1) cv.putText(video_frame.matrix, f'Segment time: {int(video_ts_ms)} ms', (20, 40), cv.FONT_HERSHEY_SIMPLEX, 1, (255, 255, 255), 1, cv.LINE_AA) - if args.window: - - # Close window using 'Esc' key - if cv.waitKey(1) == 27: - break + # Close window using 'Esc' key + if cv.waitKey(1) == 27: + break - cv.imshow(f'Segment {tobii_segment.get_id()} video', video_frame.matrix) + cv.imshow(f'Segment {tobii_segment.get_id()} video', video_frame.matrix) # Update Progress Bar progress = video_ts_ms - int(args.time_range[0] * 1e3) |