This commit is contained in:
AdamRashid96 committed 2023-06-21 02:53:14 +00:00
1 parent 0e2edd9b34
commit 7130b34790
1 file changed
-1
-1
View File
@@ -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