From 2df0a95f85bc415d73e8e5e5992031150ea7e8a6 Mon Sep 17 00:00:00 2001
From: Pablo Martin <pablo@develagora.com>
Date: Mon, 30 Dec 2024 10:06:56 +0100
Subject: [PATCH] reduce ram usage using fisheye as camera model (#521)

---
 examples/datasets/colmap.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/examples/datasets/colmap.py b/examples/datasets/colmap.py
index 938bad26..8d47a969 100644
--- 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(