diff --git a/README-pandora.md b/README-pandora.md index ca2ab2a41..bca469744 100644 --- a/README-pandora.md +++ b/README-pandora.md @@ -155,7 +155,7 @@ ns-process-data images --data /mnt/data/garden/images --output-dir /mnt/data/gar 1. Run step 1 of lantern-nerfacto. ``` - ns-train lantern-nerfacto --data /mnt/data/scene/scene_ns/ --viewer.websocket-port 8008 --pipeline.datamanager.train-num-images-to-sample-from 1800 --pipeline.model.lantern_steps 1 --pipeline.datamanager.pixel-sampler.lantern_steps 1 --max-num-iterations 60000 + ns-train PanDORA --data /mnt/data/scene/scene_ns/ --viewer.websocket-port 8008 --pipeline.datamanager.train-num-images-to-sample-from 1800 --pipeline.model.lantern_steps 1 --pipeline.datamanager.pixel-sampler.lantern_steps 1 --max-num-iterations 60000 ``` 1.1. For better alignment between the well and the fast exposed images, run the following command. @@ -166,7 +166,7 @@ ns-process-data images --data /mnt/data/garden/images --output-dir /mnt/data/gar 2. Run step 2 of lantern-nerfacto. ``` - ns-train lantern-nerfacto --data /mnt/data/scene/scene_ns/ --viewer.websocket-port 8008 --pipeline.datamanager.train-num-images-to-sample-from 1800 --pipeline.model.lantern_steps 2 --pipeline.datamanager.pixel-sampler.lantern_steps 2 --load-dir /mnt/workspace/lantern/nerfstudio-HDR/outputs/scene_ns/lantern-nerfacto/2024-06-10_155559/nerfstudio_models --pipeline.model.apply_mu_law False --max-num-iterations 120000 + ns-train PanDORA --data /mnt/data/scene/scene_ns/ --viewer.websocket-port 8008 --pipeline.datamanager.train-num-images-to-sample-from 1800 --pipeline.model.lantern_steps 2 --pipeline.datamanager.pixel-sampler.lantern_steps 2 --load-dir /mnt/workspace/lantern/nerfstudio-HDR/outputs/scene_ns/lantern-nerfacto/2024-06-10_155559/nerfstudio_models --pipeline.model.apply_mu_law False --max-num-iterations 120000 ``` ## 4. View results diff --git a/nerfstudio/configs/method_configs.py b/nerfstudio/configs/method_configs.py index 15ee77809..a5988d9e8 100644 --- a/nerfstudio/configs/method_configs.py +++ b/nerfstudio/configs/method_configs.py @@ -78,9 +78,9 @@ method_configs: Dict[str, TrainerConfig] = {} descriptions = { "nerfacto": "Recommended real-time model tuned for real captures. This model will be continually updated.", "nerfacto-HDR": "Nerfacto that take HDR as input, and output HDR field.", - "lantern-nerfacto": "Nerfacto that train on two exposures.", + "PanDORA": "Nerfacto that train on two exposures.", "panoHDR-nerfacto": "Nerfacto that take HDR as input, and output HDR field.", - # "lantern-nerfacto": "Nerfacto that take HDR as input, and output HDR field.", + # "PanDORA": "Nerfacto that take HDR as input, and output HDR field.", "hdr-nerfacto": "HDR-Nerfacto, adapted to the Nerfacto model.", "hdr-nerfacto-wo-crf": "HDR-Nerfacto without CRF, adapted to the Nerfacto model.", "depth-nerfacto": "Nerfacto with depth supervision.", @@ -131,7 +131,7 @@ method_configs["nerfacto"] = TrainerConfig( vis="viewer", ) -method_configs["lantern-nerfacto"] = Get_lantern_config() +method_configs["PanDORA"] = Get_lantern_config() method_configs["panoHDR-nerfacto"] = Get_PanoHDR_Nerfacto_config() method_configs["hdr-nerfacto"] = get_hdr_nerfacto_config(use_crf=True, clip_before_accumulation=False) diff --git a/nerfstudio/lantern/config.py b/nerfstudio/lantern/config.py index a93b31fb8..d3401b307 100644 --- a/nerfstudio/lantern/config.py +++ b/nerfstudio/lantern/config.py @@ -20,7 +20,7 @@ from nerfstudio.pipelines.base_pipeline import VanillaPipelineConfig def Get_lantern_config(): return TrainerConfig( - method_name="lantern-nerfacto", + method_name="PanDORA", steps_per_eval_batch=500, steps_per_save=2000, max_num_iterations=100000,