Skip to content

sample pixels from all data in variable-resolution batches#2772

Merged
brentyi merged 10 commits intonerfstudio-project:mainfrom
decrispell:aux_sampling
Apr 24, 2025
Merged

sample pixels from all data in variable-resolution batches#2772
brentyi merged 10 commits intonerfstudio-project:mainfrom
decrispell:aux_sampling

Conversation

@decrispell
Copy link
Copy Markdown
Contributor

This PR bring the behavior of collate_image_dataset_batch_list in line with that of collate_image_dataset_batch by sampling pixels from all data in the batch, not just "image".

@decrispell
Copy link
Copy Markdown
Contributor Author

@brentyi I believe this (old) MR is still needed to fix the collate_image_dataset_batch_list() function for batches that contain any per-pixel data other than "image" and "depth_image". If the batch contains any other image-like data, it will crash here since value will be a list, not a tensor.

Any chance this could get merged so we don't have to continue keeping it up to date with main in parallel?

Copy link
Copy Markdown
Collaborator

@brentyi brentyi left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks @decrispell for the bump! I did a pass and this looks reasonable to me.

@brentyi brentyi merged commit 5003d0e into nerfstudio-project:main Apr 24, 2025
3 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants