Spaces:
Paused
Paused
Update run_train_infer.sh
Browse files- run_train_infer.sh +6 -3
run_train_infer.sh
CHANGED
|
@@ -37,14 +37,16 @@ for DATASET in "${DATASETS[@]}"; do
|
|
| 37 |
MODEL_PATH=./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 |
--img_base_path ${SOURCE_PATH} \
|
| 42 |
--n_views ${N_VIEW} \
|
| 43 |
--focal_avg \
|
| 44 |
"
|
| 45 |
|
| 46 |
# # ----- (2) Train: jointly optimize pose -----
|
| 47 |
-
CMD_T="CUDA_VISIBLE_DEVICES=${GPU_ID} python -W ignore ./train_joint.py \
|
|
|
|
| 48 |
-s ${SOURCE_PATH} \
|
| 49 |
-m ${MODEL_PATH} \
|
| 50 |
--n_views ${N_VIEW} \
|
|
@@ -54,7 +56,8 @@ for DATASET in "${DATASETS[@]}"; do
|
|
| 54 |
"
|
| 55 |
|
| 56 |
# ----- (3) Render interpolated pose & output video -----
|
| 57 |
-
CMD_RI="CUDA_VISIBLE_DEVICES=${GPU_ID} python -W ignore ./render_by_interp.py \
|
|
|
|
| 58 |
-s ${SOURCE_PATH} \
|
| 59 |
-m ${MODEL_PATH} \
|
| 60 |
--n_views ${N_VIEW} \
|
|
|
|
| 37 |
MODEL_PATH=./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} \
|
|
|
|
| 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} \
|