Files
nerfstudio/nerf_training_lab_downstairs.sh
T
2024-05-18 19:57:48 -04:00

25 lines
1.6 KiB
Bash
Executable File

#!/bin/bash
echo "Evaluation for linearization then nerf pipeline"
dataset_name=lab_downstairs_ns
method_name=lantern-nerfacto
root_dir=/mnt/data/nerfstudio_ds/real_data/lab_downstairs_data
dataset_dir=$root_dir/$dataset_name/
opensfm_dir=$root_dir/lab_downstairs_openSFM
opensfm_file=$opensfm_dir/reconstruction.json
echo "Step one of training"
ns-train lantern-nerfacto --data $dataset_dir --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 \
--pipeline.datamanager.camera-optimizer.mode off --viewer.quit-on-train-completion True --output-dir outputs --timestamp step_1
echo "Step two of training"
#ns-train lantern-nerfacto --data $dataset_dir --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 --pipeline.model.apply_mu_law False \
#--pipeline.datamanager.camera-optimizer.mode off --viewer.quit-on-train-completion True --load-dir outputs/$dataset_name/lantern-nerfacto/step_1/nerfstudio_models \
#--output-dir outputs --timestamp step_2
ns-train lantern-nerfacto --data $dataset_dir --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 \
--pipeline.datamanager.camera-optimizer.mode off --load-dir outputs/$dataset_name/lantern-nerfacto/step_1/nerfstudio_models \
--output-dir outputs --timestamp step_2