Fix mask shape for Fisheye624 cameras (nerfstudio-project#2813)

ArpegorPSGH · Jun 22, 2024 · cd3fd74 · cd3fd74
1 parent de85d8d
commit cd3fd74
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")