diff --git a/nerfstudio/data/pixel_samplers.py b/nerfstudio/data/pixel_samplers.py index 780c4b44f..e89d287b9 100644 --- a/nerfstudio/data/pixel_samplers.py +++ b/nerfstudio/data/pixel_samplers.py @@ -223,7 +223,6 @@ class PixelSampler: image_height, image_width, _ = batch["image"][i].shape if i == num_images - 1: num_rays_in_batch = num_rays_per_batch - (num_images - 1) * num_rays_in_batch - indices = self.sample_method(num_rays_in_batch, 1, image_height, image_width, device=device) if self.config.is_equirectangular: indices = self.sample_method_equirectangular( num_rays_in_batch, 1, image_height, image_width, device=device