Updated main.py

 src/model.py
 src/mouse_controller.py

Signed-off-by: Mpho Mphego <mpho112@gmail.com>

Author: mmphego

main.py

@@ -198,7 +198,6 @@ def main(args):
                     head_pose_estimation.show_text(frame, head_pose_angles)
                     gaze_estimation.show_text(frame, gaze_vector)
 
-                print(f"gaze_vector: x: {gaze_vector['x']:.2f}, y: {gaze_vector['y']:.2f}")
                 if args.enable_mouse:
                     mouse_controller.move(gaze_vector["x"], gaze_vector["y"])
 

src/model.py

@@ -257,18 +257,22 @@ def preprocess_output(self, inference_results, image, show_bbox=False):
         eye_size = 10
 
         left_eye_x_coord = int(eyes_coords[0] * w)
+        # left eye offset of face
         left_eye_xmin = left_eye_x_coord - eye_size
         left_eye_xmax = left_eye_x_coord + eye_size
 
         left_eye_y_coord = int(eyes_coords[1] * h)
+        # left eye offset of face
         left_eye_ymin = left_eye_y_coord - eye_size
         left_eye_ymax = left_eye_y_coord + eye_size
 
         right_eye_x_coord = int(eyes_coords[2] * w)
+        # right eye offset of face
         right_eye_xmin = right_eye_x_coord - eye_size
         right_eye_xmax = right_eye_x_coord + eye_size
 
         right_eye_y_coord = int(eyes_coords[3] * h)
+        # right eye offset of face
         right_eye_ymin = right_eye_y_coord - eye_size
         right_eye_ymax = right_eye_y_coord + eye_size
 

src/mouse_controller.py

@@ -25,14 +25,14 @@ def move(self, x, y):
         try:
             start_pos = pyautogui.position()
             pyautogui.moveRel(
-                -x * self.precision, y * self.precision, duration=self.speed
+                -x * self.precision/2., y * self.precision/2., duration=self.speed
             )
             end_pos = pyautogui.position()
-            logger.info(f"Mouse -> start_pos: {start_pos}, end_pos: {end_pos}")
+            logger.debug(f"Mouse -> start_pos: {start_pos}, end_pos: {end_pos}")
         except pyautogui.FailSafeException:
             logger.exception(f"Position: {x}, {y} are out of the screen")
             pyautogui.moveRel(
-                x * self.precision, -1 * y * self.precision, duration=self.speed
+                -x * self.precision, y * self.precision, duration=self.speed
             )
     def left_click(self):
         pass