This commit is contained in:
Ethan Weber committed 2022-04-23 19:51:05 +00:00
1 parent daf5a6251a
commit 11d9c7271a
1 file changed
+2 -2
+2 -2
View File
@@ -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,
}