diff --git a/gaze_tracking/eye.py b/gaze_tracking/eye.py index fe8fa64..a3efdc0 100644 --- a/gaze_tracking/eye.py +++ b/gaze_tracking/eye.py @@ -18,6 +18,7 @@ def __init__(self, original_frame, landmarks, side, calibration): self.origin = None self.center = None self.pupil = None + self.landmark_points = None self._analyze(original_frame, landmarks, side, calibration) @@ -43,6 +44,7 @@ def _isolate(self, frame, landmarks, points): """ region = np.array([(landmarks.part(point).x, landmarks.part(point).y) for point in points]) region = region.astype(np.int32) + self.landmark_points = region # Applying a mask to get only the eye height, width = frame.shape[:2]