SW777SW commited on
Commit
d707432
·
verified ·
1 Parent(s): 1b597ee

Delete run_train_infer.sh

Browse files
Files changed (1) hide show
  1. run_train_infer.sh +0 -79
run_train_infer.sh DELETED
@@ -1,79 +0,0 @@
1
- #! /bin/bash
2
-
3
- GPU_ID=1
4
- DATA_ROOT_DIR="/app/data"
5
- DATASETS=(
6
- # TT
7
- sora
8
- # mars
9
- )
10
-
11
- SCENES=(
12
- # Family
13
- # Barn
14
- # Francis
15
- # Horse
16
- # Ignatius
17
- santorini
18
- )
19
-
20
- N_VIEWS=(
21
- 3
22
- # 5
23
- # 9
24
- # 12
25
- )
26
-
27
- # increase iteration to get better metrics (e.g. gs_train_iter=5000)
28
- gs_train_iter=1000
29
- pose_lr=1x
30
-
31
- for DATASET in "${DATASETS[@]}"; do
32
- for SCENE in "${SCENES[@]}"; do
33
- for N_VIEW in "${N_VIEWS[@]}"; do
34
-
35
- # SOURCE_PATH must be Absolute path
36
- SOURCE_PATH=${DATA_ROOT_DIR}/${DATASET}/${SCENE}/${N_VIEW}_views
37
- MODEL_PATH=/app/output/infer/${DATASET}/${SCENE}/${N_VIEW}_views_${gs_train_iter}Iter_${pose_lr}PoseLR/
38
-
39
- # # ----- (1) Dust3r_coarse_geometric_initialization -----
40
- # CMD_D1="CUDA_VISIBLE_DEVICES=${GPU_ID} python -W ignore ./coarse_init_infer.py \
41
- CMD_D1="python -W ignore ./coarse_init_infer.py \
42
- --img_base_path ${SOURCE_PATH} \
43
- --n_views ${N_VIEW} \
44
- --focal_avg \
45
- "
46
-
47
- # # ----- (2) Train: jointly optimize pose -----
48
- # CMD_T="CUDA_VISIBLE_DEVICES=${GPU_ID} python -W ignore ./train_joint.py \
49
- CMD_T="python -W ignore ./train_joint.py \
50
- -s ${SOURCE_PATH} \
51
- -m ${MODEL_PATH} \
52
- --n_views ${N_VIEW} \
53
- --scene ${SCENE} \
54
- --iter ${gs_train_iter} \
55
- --optim_pose \
56
- "
57
-
58
- # ----- (3) Render interpolated pose & output video -----
59
- # CMD_RI="CUDA_VISIBLE_DEVICES=${GPU_ID} python -W ignore ./render_by_interp.py \
60
- CMD_RI="python -W ignore ./render_by_interp.py \
61
- -s ${SOURCE_PATH} \
62
- -m ${MODEL_PATH} \
63
- --n_views ${N_VIEW} \
64
- --scene ${SCENE} \
65
- --iter ${gs_train_iter} \
66
- --eval \
67
- --get_video \
68
- "
69
-
70
-
71
- echo "========= ${SCENE}: Dust3r_coarse_geometric_initialization ========="
72
- eval $CMD_D1
73
- echo "========= ${SCENE}: Train: jointly optimize pose ========="
74
- eval $CMD_T
75
- echo "========= ${SCENE}: Render interpolated pose & output video ========="
76
- eval $CMD_RI
77
- done
78
- done
79
- done