Fix datamanager dataparser_outputs (#1123)

* Fix datamanager dataparser_outputs

* Cast to float

Author: tancik

nerfstudio/data/datamanagers/base_datamanager.py

@@ -314,14 +314,14 @@ def __init__(
         self.test_mode = test_mode
         self.test_split = "test" if test_mode in ["test", "inference"] else "val"
         self.dataparser = self.config.dataparser.setup()
+        self.train_dataparser_outputs = self.dataparser.get_dataparser_outputs(split="train")
 
         self.train_dataset = self.create_train_dataset()
         self.eval_dataset = self.create_eval_dataset()
         super().__init__()
 
     def create_train_dataset(self) -> InputDataset:
         """Sets up the data loaders for training"""
-        self.train_dataparser_outputs = self.dataparser.get_dataparser_outputs(split="train")
         return InputDataset(
             dataparser_outputs=self.train_dataparser_outputs,
             scale_factor=self.config.camera_res_scale_factor,

nerfstudio/data/datamanagers/semantic_datamanager.py

@@ -42,7 +42,7 @@ class SemanticDataManager(VanillaDataManager):  # pylint: disable=abstract-metho
 
     def create_train_dataset(self) -> SemanticDataset:
         return SemanticDataset(
-            dataparser_outputs=self.dataparser.get_dataparser_outputs(split="train"),
+            dataparser_outputs=self.train_dataparser_outputs,
             scale_factor=self.config.camera_res_scale_factor,
         )
 

nerfstudio/data/dataparsers/base_dataparser.py

@@ -82,7 +82,7 @@ def save_dataparser_transform(self, path: Path):
         """
         data = {
             "transform": self.dataparser_transform.tolist(),
-            "scale": self.dataparser_scale,
+            "scale": float(self.dataparser_scale),
         }
         if not path.parent.exists():
             path.parent.mkdir(parents=True)