Fix undistorted eval image height/width

nerfstudio-project · Jan 18, 2024 · 736ce81 · 736ce81
1 parent 51b54eb
commit 736ce81
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/nerfstudio/data/datamanagers/full_images_datamanager.py b/nerfstudio/data/datamanagers/full_images_datamanager.py
@@ -171,6 +171,8 @@ def cache_images(self, cache_images_option):
             self.eval_dataset.cameras.fy[i] = float(K[1, 1])
             self.eval_dataset.cameras.cx[i] = float(K[0, 2])
             self.eval_dataset.cameras.cy[i] = float(K[1, 2])
+            self.eval_dataset.cameras.width[i] = image.shape[1]
+            self.eval_dataset.cameras.height[i] = image.shape[0]
 
         if cache_images_option == "gpu":
             for cache in cached_train: