fix: fixed leap visualization with ROI cropping

EyeTrackVR · Nov 13, 2023 · 408afdd · 408afdd
1 parent 0f475fc
commit 408afdd
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/TrackingBackend/app/algorithms/leap.py b/TrackingBackend/app/algorithms/leap.py
@@ -27,8 +27,8 @@
 ------------------------------------------------------------------------------------------------------
 """
 
-import math
 import cv2
+import math
 import numpy as np
 import onnxruntime as rt
 from app.types import EyeData
@@ -97,9 +97,7 @@ def run_model(self, frame: MatLike) -> np.ndarray:
         return pre_landmark
 
     def draw_landmarks(self, frame: MatLike, landmarks: np.ndarray) -> None:
-        width, height = frame.shape[:2]
-        height -= 112
-        width += 112
+        height, width = frame.shape[:2]
 
         for point in landmarks:
             x, y = point