diff options
Diffstat (limited to 'src/argaze/utils/demo_ar_features_run.py')
-rw-r--r-- | src/argaze/utils/demo_ar_features_run.py | 18 |
1 files changed, 2 insertions, 16 deletions
diff --git a/src/argaze/utils/demo_ar_features_run.py b/src/argaze/utils/demo_ar_features_run.py index 5d03fd9..22b0d89 100644 --- a/src/argaze/utils/demo_ar_features_run.py +++ b/src/argaze/utils/demo_ar_features_run.py @@ -83,25 +83,11 @@ def main(): # Display environment cv2.imshow(ar_environment.name, video_image) - # Draw and display each frames + # Draw and display each frames in separate window for scene_name, frame_name, frame in ar_environment.frames: - image = frame.background.copy() - - frame.aoi_2d_scene.draw(image, color=(255, 255, 255)) - frame.current_gaze_position.draw(image, color=(255, 255, 255)) - - frame.current_gaze_movement.draw(image, color=(0, 255, 255)) - frame.current_gaze_movement.draw_positions(image) - - # Check frame fixation - if GazeFeatures.is_fixation(frame.current_gaze_movement): - - # Draw looked AOI - frame.aoi_2d_scene.draw_circlecast(image, frame.current_gaze_movement.focus, frame.current_gaze_movement.deviation_max, base_color=(0, 0, 0), matching_color=(255, 255, 255)) - # Display frame - cv2.imshow(f'{scene_name}:{frame_name}', image) + cv2.imshow(f'{scene_name}:{frame_name}', frame.image) # Stop by pressing 'Esc' key if cv2.waitKey(10) == 27: |