feat: do rectification on gpu

UCSD-E4E · Sep 14, 2024 · f871659 · f871659
1 parent 0b7ff3a
commit f871659
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyfishsensedev/image/image_rectifier.py b/pyfishsensedev/image/image_rectifier.py
@@ -12,10 +12,10 @@ def __init__(self, lens_calibration: LensCalibration):
 
     def rectify(self, img: np.ndarray) -> np.ndarray:
         return cv2.undistort(
-            img,
+            cv2.UMat(img),
             self._lens_calibration.camera_matrix,
             self._lens_calibration.distortion_coefficients,
-        )
+        ).get()
 
 
 if __name__ == "__main__":