fix wrong image_ids in distributed mode

zerolover · zerolover · commit 4175fc0557b4 · 2025-06-27T14:08:11.000+08:00
diff --git a/gsplat/rendering.py b/gsplat/rendering.py
@@ -568,6 +568,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
`@@ -568,6 +568,10 @@ def reshape_view(C: int, world_view: torch.Tensor, N_world: list) -> torch.Tenso`
`568`	`568`	`output_splits=collected_splits,`
`569`	`569`	`)`
`570`	`570`
	`571`	`+ # since distributed mode does not support batch dimensions, it is safe`
	`572`	`+ # to set image_ids to camera_ids.`
	`573`	`+ image_ids = camera_ids`
	`574`	`+`
`571`	`575`	`# Silently change C from global #Cameras to local #Cameras.`
`572`	`576`	`C = C_world[world_rank]`
`573`	`577`