mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-11 18:44:21 +08:00
black
This commit is contained in:
1 parent
daf5a6251a
commit
11d9c7271a
1 file changed
+2
-2
@@ -35,8 +35,8 @@ def collate_batch(batch_list, num_rays_per_batch, keep_full_image: bool = False)
|
||||
local_indices = indices.clone()
|
||||
indices[:, 0] = batch["image_idx"][c]
|
||||
collated_batch = {
|
||||
"local_indices": local_indices, # local to the batch returned
|
||||
"indices": indices, # with the abs camera indices
|
||||
"local_indices": local_indices, # local to the batch returned
|
||||
"indices": indices, # with the abs camera indices
|
||||
"pixels": pixels,
|
||||
"mask": mask,
|
||||
}
|
||||
|
||||
Reference in new issue
Block a user