mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-12 19:04:03 +08:00
42 lines
2.3 KiB
Bash
Executable File
42 lines
2.3 KiB
Bash
Executable File
#!/bin/bash
|
|
echo "Evaluation for linearization then nerf pipeline"
|
|
|
|
dataset_name=meeting_room_ns
|
|
method_name=lantern-nerfacto
|
|
root_dir=/mnt/data/nerfstudio_ds/real_data/meeting_room_openSFM
|
|
dataset_dir=$root_dir/$dataset_name/
|
|
opensfm_dir=$root_dir/meeting_room_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.model.apply_mu_law False \
|
|
#--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
|
|
|
|
checkpoint_dir=outputs/$dataset_name/lantern-nerfacto/step_2/
|
|
output_dir=renders/$dataset_name/$method_name/ours_results
|
|
|
|
echo "Creating the test set"
|
|
ns-process-data lantern-GT-HDR --data $dataset_dir --output-dir $dataset_dir --metadata $opensfm_file --checkpoint $checkpoint_dir
|
|
|
|
echo "Rendering for the well exposed"
|
|
ns-render camera-path --load-config $checkpoint_dir/config.yml --camera-path-filename $dataset_dir/GT_transforms.json --output-path $output_dir/well-expo/ --output-format images
|
|
|
|
echo "Rendering for the fast exposure"
|
|
ns-render camera-path --load-config $checkpoint_dir/config.yml --camera-path-filename $dataset_dir/GT_transforms.json --output-path $output_dir/fast-expo/ --output-format images --rendered_output_names rgb_fast
|
|
|
|
#conda init bash
|
|
conda activate everlight_new
|
|
|
|
echo "Combine two exposures"
|
|
python lantern_scripts/combine.py --well_dir $output_dir/well-expo/ --fast_dir $output_dir/fast-expo/ --out_dir $output_dir/pano/ --do_linearization --experiment_location $root_dir
|
|
|
|
conda activate nerfstudio-HDR
|
|
|
|
blender --background lantern_scripts/scene_new.blend -P lantern_scripts/hdr_blender.py -- $output_dir/pano/ $output_dir/render/ |