Fix precision loss of lms.z variable

cvjena · Mar 16, 2023 · db95ec6 · db95ec6
1 parent b6432ba
commit db95ec6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/face_projection/core.py b/src/face_projection/core.py
@@ -91,7 +91,8 @@ def get_landmarks(self, face_img: np.ndarray):
         if results.multi_face_landmarks:
             lms = results.multi_face_landmarks[0].landmark
             for i in range(468):
-                self.__landmarks[i, :] = int(lms[i].x * w), int(lms[i].y * h), lms[i].z
+                # scale z by w to net be removed by conversion to int
+                self.__landmarks[i, :] = int(lms[i].x * w), int(lms[i].y * h), int(lms[i].z * w)
 
     def apply(
         self,