mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-13 03:14:00 +08:00
bug fix
This commit is contained in:
1 parent
0e2edd9b34
commit
7130b34790
1 file changed
-1
@@ -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
|
||||
|
||||
Reference in new issue
Block a user