chore: reduce ram usage using fisheye as camera model

nerfstudio-project · Dec 23, 2024 · 63028c3 · 63028c3
1 parent ff2da2e
commit 63028c3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/datasets/colmap.py b/examples/datasets/colmap.py
@@ -266,8 +266,8 @@ def __init__(
                     + params[2] * theta**6
                     + params[3] * theta**8
                 )
-                mapx = fx * x1 * r + width // 2
-                mapy = fy * y1 * r + height // 2
+                mapx = (fx * x1 * r + width // 2).astype(np.float32)
+                mapy = (fy * y1 * r + height // 2).astype(np.float32)
 
                 # Use mask to define ROI
                 mask = np.logical_and(