diff options
| author | Théo de la Hogue | 2023-10-17 18:49:34 +0200 |
|---|---|---|
| committer | Théo de la Hogue | 2023-10-17 18:49:34 +0200 |
| commit | 4575911ec8a04d7d228109acbcb58b448b72fb3b (patch) | |
| tree | 651476dd09a36902b45e6be49478b888f4e15f8d /src | |
| parent | 07fb4ce51650e9b0edaf1b9ebc01c1b9589c9a54 (diff) | |
| download | argaze-4575911ec8a04d7d228109acbcb58b448b72fb3b.zip argaze-4575911ec8a04d7d228109acbcb58b448b72fb3b.tar.gz argaze-4575911ec8a04d7d228109acbcb58b448b72fb3b.tar.bz2 argaze-4575911ec8a04d7d228109acbcb58b448b72fb3b.tar.xz | |
Adding draw_gaze_position_calibrator as image_parameters ArFrame option.
Diffstat (limited to 'src')
| -rw-r--r-- | src/argaze/ArFeatures.py | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/src/argaze/ArFeatures.py b/src/argaze/ArFeatures.py index 2e278ea..13b952d 100644 --- a/src/argaze/ArFeatures.py +++ b/src/argaze/ArFeatures.py @@ -993,13 +993,14 @@ class ArFrame(): # Return look data return self.__gaze_position, identified_gaze_movement, scan_step_analysis, layer_analysis, execution_times, exception - def __image(self, background_weight: float = None, heatmap_weight: float = None, draw_scan_path: dict = None, draw_layers: dict = None, draw_gaze_positions: dict = None, draw_fixations: dict = None, draw_saccades: dict = None) -> numpy.array: + def __image(self, background_weight: float = None, heatmap_weight: float = None, draw_gaze_position_calibrator: dict = None, draw_scan_path: dict = None, draw_layers: dict = None, draw_gaze_positions: dict = None, draw_fixations: dict = None, draw_saccades: dict = None) -> numpy.array: """ Get background image with overlaid visualisations. Parameters: background_weight: weight of background overlay heatmap_weight: weight of heatmap overlay + draw_gaze_position_calibrator: [GazeFeatures.GazePositionCalibrator.draw](argaze.md/#argaze.GazeFeatures.GazePositionCalibrator.draw) parameters (if None, nothing is drawn) draw_scan_path: [GazeFeatures.ScanPath.draw](argaze.md/#argaze.GazeFeatures.ScanPath.draw) parameters (if None, no scan path is drawn) draw_layers: dictionary of [ArLayer.draw](argaze.md/#argaze.ArFeatures.ArLayer.draw) parameters per layer (if None, no layer is drawn) draw_gaze_positions: [GazeFeatures.GazePosition.draw](argaze.md/#argaze.GazeFeatures.GazePosition.draw) parameters (if None, no gaze position is drawn) @@ -1032,6 +1033,11 @@ class ArFrame(): image = numpy.full((self.size[1], self.size[0], 3), 0).astype(numpy.uint8) + # Draw gaze position calibrator + if draw_gaze_position_calibrator is not None: + + self.gaze_position_calibrator.draw(image, size=self.size, **draw_gaze_position_calibrator) + # Draw scan path if required if draw_scan_path is not None and self.scan_path is not None: |