Convert points to float before cv2.undistortPoints

talmolab · Nov 27, 2024 · 250a8c5 · 250a8c5
1 parent 6240f0d
commit 250a8c5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sleap_io/model/camera.py b/sleap_io/model/camera.py
@@ -248,7 +248,7 @@ def undistort_points(self, points: np.ndarray) -> np.ndarray:
         """
         shape = points.shape
         points = points.reshape(-1, 1, 2)
-        out = cv2.undistortPoints(points, self.matrix, self.dist)
+        out = cv2.undistortPoints(points.astype("float64"), self.matrix, self.dist)
         return out.reshape(shape)
 
     def project(self, points: np.ndarray) -> np.ndarray: