diff options
Diffstat (limited to 'src/argaze/GazeAnalysis/DispersionThresholdIdentification.py')
-rw-r--r-- | src/argaze/GazeAnalysis/DispersionThresholdIdentification.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/argaze/GazeAnalysis/DispersionThresholdIdentification.py b/src/argaze/GazeAnalysis/DispersionThresholdIdentification.py index 9d5a8d7..c79b4fc 100644 --- a/src/argaze/GazeAnalysis/DispersionThresholdIdentification.py +++ b/src/argaze/GazeAnalysis/DispersionThresholdIdentification.py @@ -78,7 +78,7 @@ class Fixation(GazeFeatures.Fixation): return self - def draw(self, frame, color=(127, 127, 127), border_color=(255, 255, 255)): + def draw(self, frame: numpy.array, color=(127, 127, 127), border_color=(255, 255, 255)): """Draw fixation into frame.""" cv2.circle(frame, (int(self.focus[0]), int(self.focus[1])), int(self.deviation_max), color, -1) @@ -91,7 +91,7 @@ class Saccade(GazeFeatures.Saccade): def __post_init__(self): super().__post_init__() - def draw(self, frame, color=(255, 255, 255)): + def draw(self, frame: numpy.array, color=(255, 255, 255)): """Draw saccade into frame.""" _, start_position = self.positions.first |