From 7130b3479009b4b694d5af7aabbdafec7f6fbb53 Mon Sep 17 00:00:00 2001 From: AdamRashid96 Date: Wed, 21 Jun 2023 02:53:14 +0000 Subject: [PATCH] bug fix --- nerfstudio/data/pixel_samplers.py | 1 - 1 file changed, 1 deletion(-) 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