moved camera opt and edit readme

AkerBP · Jan 24, 2024 · aa292e1 · aa292e1
1 parent 00387b0
commit aa292e1
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 15 deletions.
diff --git a/.gitignore b/.gitignore
@@ -2,4 +2,4 @@ data/*
 outputs/*
 seathru_nerf.egg-info/*
 .vscode/*
-**/__pycache__/*
+*/__pycache__/*
diff --git a/README.md b/README.md
@@ -100,6 +100,8 @@ To train your first subsea NeRF, please download the dataset available [here](ht
 ns-train seathru-nerf-lite --vis viewer+wandb colmap --data <path_to_dataset> --images-path images_wb
 ```
 
+(If you get an error with the colmap path, create a directory `colmap` in the dataset folder and move the `sparse` folder into it.)
+
 You can then track the training process on [wandb](https://wandb.ai/site) or watch it live via the [nerfstudio viewer](https://docs.nerf.studio/en/latest/quickstart/viewer_quickstart.html).
 
 When training is completed (takes around 1hr for the SeaThru-NeRF dataset scenes with a RTX3060 Laptop GPU), the trained model will be saved to an `./outputs/` folder.

diff --git a/seathru/seathru_config.py b/seathru/seathru_config.py
@@ -24,13 +24,6 @@
                 dataparser=NerfstudioDataParserConfig(),
                 train_num_rays_per_batch=16384,
                 eval_num_rays_per_batch=4096,
-                camera_optimizer=CameraOptimizerConfig(
-                    mode="off",
-                    optimizer=AdamOptimizerConfig(lr=6e-4, eps=1e-8, weight_decay=1e-2),
-                    scheduler=ExponentialDecaySchedulerConfig(
-                        lr_final=6e-6, max_steps=500000
-                    ),
-                ),
             ),
             model=SeathruModelConfig(eval_num_rays_per_chunk=1 << 15),
         ),
@@ -47,6 +40,13 @@
                     lr_final=1e-5, max_steps=500000, warmup_steps=1024
                 ),
             },
+            "camera_opt": {
+                "mode": "off",
+                "optimizer": AdamOptimizerConfig(lr=6e-4, eps=1e-8, weight_decay=1e-2),
+                "scheduler": ExponentialDecaySchedulerConfig(
+                        lr_final=6e-6, max_steps=500000
+                    ),
+            },
         },
         viewer=ViewerConfig(num_rays_per_chunk=1 << 15),
         vis="viewer",
@@ -67,13 +67,6 @@
                 dataparser=NerfstudioDataParserConfig(),
                 train_num_rays_per_batch=8192,
                 eval_num_rays_per_batch=4096,
-                camera_optimizer=CameraOptimizerConfig(
-                    mode="off",
-                    optimizer=AdamOptimizerConfig(lr=6e-4, eps=1e-8, weight_decay=1e-2),
-                    scheduler=ExponentialDecaySchedulerConfig(
-                        lr_final=6e-6, max_steps=500000
-                    ),
-                ),
             ),
             model=SeathruModelConfig(
                 eval_num_rays_per_chunk=1 << 15,
@@ -116,6 +109,13 @@
                     lr_final=1e-5, max_steps=500000, warmup_steps=1024
                 ),
             },
+            "camera_opt": {
+                "mode": "off",
+                "optimizer": AdamOptimizerConfig(lr=6e-4, eps=1e-8, weight_decay=1e-2),
+                "scheduler": ExponentialDecaySchedulerConfig(
+                        lr_final=6e-6, max_steps=500000
+                    ),
+            },
         },
         viewer=ViewerConfig(num_rays_per_chunk=1 << 15),
         vis="viewer",