tink/nerfstudio
4
0
Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 10:34:29 +08:00
Code

fix_mask (#1819)

Browse Source 
fix_nerfacto_mask

Co-authored-by: machenmusik <machenmusik@users.noreply.github.com>
This commit is contained in:
Yuantao Chen 2023-04-26 09:32:20 +08:00 committed by GitHub
parent 0eb0ceb3c0
commit d875f5f5b5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
nerfstudio/data/pixel_samplers.py
View File

@ -96,6 +96,7 @@ class PixelSampler: # pylint: disable=too-few-public-methods
indices = self.sample_method(num_rays_per_batch, num_images, image_height, image_width, device=device)
c, y, x = (i.flatten() for i in torch.split(indices, 1, dim=-1))
c, y, x = c.cpu(), y.cpu(), x.cpu()
collated_batch = {
key: value[c, y, x] for key, value in batch.items() if key != "image_idx" and value is not None
}