fix wrong image_ids in distributed mode

zerolover · zerolover · commit df6c3357a851 · 2025-07-09T08:47:51.000+08:00
diff --git a/gsplat/rendering.py b/gsplat/rendering.py
@@ -575,6 +575,10 @@ def reshape_view(C: int, world_view: torch.Tensor, N_world: list) -> torch.Tenso
                 output_splits=collected_splits,
             )
 
+            # since distributed mode does not support batch dimensions, it is safe
+            # to set image_ids to camera_ids.
+            image_ids = camera_ids
+
             # Silently change C from global #Cameras to local #Cameras.
             C = C_world[world_rank]
 

Original file line number	Diff line number	Diff line change
`@@ -575,6 +575,10 @@ def reshape_view(C: int, world_view: torch.Tensor, N_world: list) -> torch.Tenso`
`575`	`575`	`output_splits=collected_splits,`
`576`	`576`	`)`
`577`	`577`
	`578`	`+ # since distributed mode does not support batch dimensions, it is safe`
	`579`	`+ # to set image_ids to camera_ids.`
	`580`	`+ image_ids = camera_ids`
	`581`	`+`
`578`	`582`	`# Silently change C from global #Cameras to local #Cameras.`
`579`	`583`	`C = C_world[world_rank]`
`580`	`584`