Spaces:
Paused
Paused
Delete run_train_infer.sh
Browse files- 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
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|