diff options
| -rw-r--r-- | src/argaze/utils/demo_gaze_features_run.py | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/src/argaze/utils/demo_gaze_features_run.py b/src/argaze/utils/demo_gaze_features_run.py index 8bd0a88..2a9474f 100644 --- a/src/argaze/utils/demo_gaze_features_run.py +++ b/src/argaze/utils/demo_gaze_features_run.py @@ -49,7 +49,7 @@ def main(): window_size = (aoi_scene_image.shape[1], aoi_scene_image.shape[0]) # Project AOI scene onto Full HD screen - aoi_scene_projection = demo_scene.orthogonal_projection * window_size + aoi_scene_projection = demo_scene.aoi_scene.orthogonal_projection * window_size # Create a window to display AR environment window_name = "AOI Scene" @@ -58,8 +58,8 @@ def main(): # Init gaze processing gaze_position = GazeFeatures.GazePosition() - screen_image = AOIFeatures.AOIFrame(aoi_scene_projection['Screen'], window_size) - screen_image.heatmap_init() + heatmap = AOIFeatures.Heatmap(window_size) + heatmap.init() enable_heatmap = False clear_heatmap = False @@ -150,11 +150,11 @@ def main(): # Clear heatmap if clear_heatmap: - screen_image.heatmap_init(10 if enable_heatmap_buffer else 0) + heatmap.init(10 if enable_heatmap_buffer else 0) clear_heatmap = False # Update heatmap - screen_image.heatmap_update(gaze_position.value, sigma=0.05) + heatmap.update(gaze_position.value, sigma=0.05) else: @@ -263,7 +263,7 @@ def main(): # Draw gaze spread heatmap if enable_heatmap: - image = cv2.addWeighted(screen_image.heatmap, 0.5, image, 1., 0) + image = cv2.addWeighted(heatmap.heatmap, 0.5, image, 1., 0) else: |