tink/nerfstudio
4
0 Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-13 11:24:00 +08:00
Code Activity

fix

Browse Source
This commit is contained in:
Ethan Weber committed 2023-08-04 20:17:11 +00:00
1 parent c11a43f249
commit 02f16b07f2
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
nerfstudio/data/pixel_samplers.py
+1 -1
View File
@@ -323,7 +323,7 @@ class PatchPixelSampler(PixelSampler):
half_patch_size = int(self.config.patch_size / 2)
m = erode_mask(mask.permute(0, 3, 1, 2).float(), pixel_radius=half_patch_size)
nonzero_indices = torch.nonzero(m[:, 0], as_tuple=False).to(device)
chosen_indices = random.sample(range(len(nonzero_indices)), k=self.rays_to_sample)
chosen_indices = random.sample(range(len(nonzero_indices)), k=sub_bs)
indices = nonzero_indices[chosen_indices]
indices = (