fixing blender scene training with vanilla nerf

This commit is contained in:
Ethan Weber committed 2022-06-08 18:00:51 +00:00
1 parent a986cb0e2a
commit d16f87a957
2 files changed
+3 -3

No files matched your search

+2 -2
View File
@@ -19,7 +19,7 @@ def load_blender_data(
downscale_factor: int = 1,
alpha_color: Optional[TensorType[3]] = None,
split: str = "train",
scale_factor: float = 0.33,
scale_factor: float = 1.0,
) -> DatasetInputs:
"""Processes the a blender dataset directory.
Some of this code comes from https://github.com/yenchenlin/nerf-pytorch/blob/master/load_blender.py#L37.
@@ -60,7 +60,7 @@ def load_blender_data(
# in x,y,z order
camera_to_world[..., 3] *= scale_factor
scene_bounds = SceneBounds(aabb=torch.tensor([[-0.5, -0.5, -0.5], [0.5, 0.5, 0.5]], dtype=torch.float32))
scene_bounds = SceneBounds(aabb=torch.tensor([[-1.5, -1.5, -1.5], [1.5, 1.5, 1.5]], dtype=torch.float32))
dataset_inputs = DatasetInputs(
image_filenames=image_filenames,
+1 -1
View File
@@ -200,7 +200,7 @@ def launch(
profiler.flush_profiler(config.logging)
@hydra.main(config_path="../configs", config_name="vanilla_nerf.yaml")
@hydra.main(config_path="../configs", config_name="graph_default.yaml")
def main(config: DictConfig):
"""Main function."""
unrolled_path = os.path.join(os.getcwd(), ".hydra/config.yaml")