diff --git a/nerfstudio/data/dataparsers/nerfstudio_dataparser.py b/nerfstudio/data/dataparsers/nerfstudio_dataparser.py index b569508523..dcc4ace781 100644 --- a/nerfstudio/data/dataparsers/nerfstudio_dataparser.py +++ b/nerfstudio/data/dataparsers/nerfstudio_dataparser.py @@ -313,7 +313,9 @@ def _generate_dataparser_outputs(self, split="train"): # - applied_transform contains the transformation to saved coordinates from original data coordinates. applied_transform = None colmap_path = self.config.data / "colmap/sparse/0" - if "applied_transform" not in meta and colmap_path.exists(): + if "applied_transform" in meta: + applied_transform = torch.tensor(meta["applied_transform"], dtype=transform_matrix.dtype) + elif colmap_path.exists(): # For converting from colmap, this was the effective value of applied_transform that was being # used before we added the applied_transform field to the output dataformat. meta["applied_transform"] = [[0, 1, 0, 0], [1, 0, 0, 0], [0, 0, -1, 0]]