Fix mask shape for Fisheye624 cameras (#2813)

nerfstudio-project · Jan 24, 2024 · fb4f9f0 · fb4f9f0
1 parent 3655920
commit fb4f9f0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/nerfstudio/data/datamanagers/full_images_datamanager.py b/nerfstudio/data/datamanagers/full_images_datamanager.py
@@ -471,7 +471,8 @@ def _undistort_image(
                 borderValue=0,
             )
             / 255.0
-        ).bool()
+        ).bool()[..., None]
+        assert mask.shape == (undist_h, undist_w, 1)
         K = undist_K.numpy()
     else:
         raise NotImplementedError("Only perspective and fisheye cameras are supported")