tink/nerfstudio
4
0 Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-12 19:04:03 +08:00
Code Activity

merge

Browse Source
This commit is contained in:
Justin committed 2023-06-21 18:51:02 +00:00
2 parents e26bd4d525 f6b6ed8ca8
commit 75d1013d60
1 file changed
+3 -1
Download Patch File Download Diff File Expand all files Collapse all files
nerfstudio/cameras/camera_optimizers.py
+3 -1
View File
@@ -100,7 +100,9 @@ class CameraOptimizer(nn.Module):
assert_never(self.config.mode)
# Detach non-trainable indices by setting to identity transform
if self.non_trainable_camera_indices is not None:
outputs[0][self.non_trainable_camera_indices] = torch.eye(4, device=self.device)[:3, :4]
if self.non_trainable_camera_indices.device != self.pose_adjustment.device:
self.non_trainable_camera_indices = self.non_trainable_camera_indices.to(self.pose_adjustment.device)
outputs[0][self.non_trainable_camera_indices] = torch.eye(4, device=self.pose_adjustment.device)[:3, :4]
# Return: identity if no transforms are needed, otherwise multiply transforms together.
if len(outputs) == 0: