-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathrun.sh
160 lines (143 loc) · 4.65 KB
/
run.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
#GeoNeuS
cd code/geoneus
conf_list=("geoneus" "geoneus_jaco")
case_list=("Scannetpp_crop_scaled/0a7cc12c0e")
split_list=("split_sub5.pkl")
length=${#case_list[@]}
for conf in "${conf_list[@]}"; do
for ((i=0; i < $length; i++)); do
case=${case_list[$i]}
split=${split_list[$i]}
python exp_runner.py --mode train --conf ./confs/$conf.conf --case $case --split $split
done
done
#i2sdf
cd code/i2-sdf
conf_list=("i2-sdf" "i2-sdf+jaco")
case_list=("Scannetpp_crop_scaled/0a7cc12c0e")
split_list=("split_sub5.pkl")
length=${#case_list[@]}
for conf in "${conf_list[@]}"; do
for ((i=0; i < $length; i++)); do
case=${case_list[$i]}
split=${split_list[$i]}
python data/normalize_cameras.py -i "../../dataset/processored/${case}/cameras_sphere.npz" -o "../../dataset/processored/${case}/cameras_normalize.npz"
python main_recon.py --conf config/$conf.yml --scan_name $case --split $split --expname "${conf}/${split}/"
python main_recon.py --conf config/$conf.yml --scan_name $case --test --test_mode mesh --split $split --expname "${conf}/${split}/"
done
done
#MonoSDF
cd code/monosdf/code
conf_list=("scannet_mlp" "scannet_mlp_jaco")
case_list=("Scannetpp_crop_scaled/0a7cc12c0e")
split_list=("split_sub5.pkl")
length=${#case_list[@]}
for conf in "${conf_list[@]}"; do
for ((i=0; i < $length; i++)); do
case=${case_list[$i]}
split=${split_list[$i]}
python training/exp_runner.py --conf confs/$conf.conf --scan_id $case --split $split
done
done
#NeuralAngelo
cd code/NeuralAngelo
case_list=("0a7cc12c0e")
split_list=("split_sub5.pkl")
length=${#case_list[@]}
for ((i=0; i < $length; i++)); do
case=${case_list[$i]}
split=${split_list[$i]}
echo "Run $case, $split"
EXPERIMENT=$case
GROUP=na
NAME="${case}_${split}"
CONFIG=projects/neuralangelo/configs/${EXPERIMENT}.yaml
GPUS=1
python train.py \
--logdir=logs/${GROUP}/${NAME} \
--config=${CONFIG} \
--single_gpu \
--split $split \
# --wandb
CONTENT=$(cat logs/${GROUP}/${NAME}/latest_checkpoint.txt)
CHECKPOINT=logs/${GROUP}/${NAME}/${CONTENT}
OUTPUT_MESH=logs/${GROUP}/${NAME}/150000.ply
CONFIG=logs/${GROUP}/${NAME}/config.yaml
RESOLUTION=512
BLOCK_RES=128
GPUS=1
python projects/neuralangelo/scripts/extract_mesh.py \
--config=${CONFIG} \
--checkpoint=${CHECKPOINT} \
--output_file=${OUTPUT_MESH} \
--resolution=${RESOLUTION} \
--block_res=${BLOCK_RES} \
--single_gpu
case="${case}_jaco"
split=${split_list[$i]}
echo "Run $case, $split"
EXPERIMENT=$case
GROUP=na
NAME="${case}_${split}"
CONFIG=projects/neuralangelo/configs/${EXPERIMENT}.yaml
GPUS=1
python train.py \
--logdir=logs/${GROUP}/${NAME} \
--config=${CONFIG} \
--single_gpu \
--split $split \
# --wandb
CONTENT=$(cat logs/${GROUP}/${NAME}/latest_checkpoint.txt)
CHECKPOINT=logs/${GROUP}/${NAME}/${CONTENT}
OUTPUT_MESH=logs/${GROUP}/${NAME}/150000.ply
CONFIG=logs/${GROUP}/${NAME}/config.yaml
RESOLUTION=512
BLOCK_RES=128
GPUS=1
python projects/neuralangelo/scripts/extract_mesh.py \
--config=${CONFIG} \
--checkpoint=${CHECKPOINT} \
--output_file=${OUTPUT_MESH} \
--resolution=${RESOLUTION} \
--block_res=${BLOCK_RES} \
--single_gpu
done
#Neuris
cd code/monosdf/code
conf_list=("neuris" "neuris_jaco")
case_list=("Scannetpp_crop_scaled/0a7cc12c0e")
split_list=("split_sub5.pkl")
length=${#case_list[@]}
for conf in "${conf_list[@]}"; do
for ((i=0; i < $length; i++)); do
case=${case_list[$i]}
split=${split_list[$i]}
python exp_runner.py --mode train --conf ./confs/$conf.conf --case $case --split $split
done
done
#NeuS, It's involved by neuris
cd code/monosdf/code
conf_list=("neus" "neus_jaco")
case_list=("Scannetpp_crop_scaled/0a7cc12c0e")
split_list=("split_sub5.pkl")
length=${#case_list[@]}
for conf in "${conf_list[@]}"; do
for ((i=0; i < $length; i++)); do
case=${case_list[$i]}
split=${split_list[$i]}
python exp_runner.py --mode train --conf ./confs/$conf.conf --case $case --split $split
done
done
#Volsdf
cd code/volsdf/code
conf_list=("volsdf" "volsdf_jaco")
case_list=("Scannetpp_crop_scaled/0a7cc12c0e")
split_list=("split_sub5.pkl")
length=${#case_list[@]}
for conf in "${conf_list[@]}"; do
for ((i=0; i < $length; i++)); do
case=${case_list[$i]}
split=${split_list[$i]}
python training/exp_runner.py --conf ./confs/$conf.conf --scan_id $case --split $split --expname "${conf}/${split}/"
done
done