misc: add scripts

Author: JasonLSC

.gitignore

@@ -145,4 +145,7 @@ figs
 gsc_utils
 examples/benchmarks/compression/mpeg_exp
 examples/benchmarks/compression/rebuttal
-examples/benchmarks/viewer
+examples/benchmarks/viewer
+
+examples/readme_on_STG.md
+examples/benchmarks/stg/old

examples/benchmarks/stg/final/stg_compression_rd_lambda_0.02.sh

@@ -0,0 +1,57 @@
+SCENE_DIR="data/neural_3d"
+SCENE_LIST="coffee_martini cook_spinach cut_roasted_beef flame_salmon_1 flame_steak sear_steak"
+#  
+
+RESULT_DIR="results/stg_neu3d_compression_final_lambda_0.02_1112"
+
+NUM_FRAME=300
+
+run_single_scene() {
+    local GPU_ID=$1
+    local SCENE=$2
+
+    echo "Running $SCENE"
+
+    CUDA_VISIBLE_DEVICES=$GPU_ID python simple_trainer_STG.py compression_sim \
+        --model_path $RESULT_DIR/$SCENE/ \
+        --data_dir $SCENE_DIR/$SCENE/colmap_0 \
+        --result_dir $RESULT_DIR/$SCENE/ \
+        --duration $NUM_FRAME \
+        --entropy_model_opt \
+        --rd_lambda 0.02
+        # --profiler_enabled
+    
+    CUDA_VISIBLE_DEVICES=$GPU_ID python simple_trainer_STG.py default \
+        --model_path $RESULT_DIR/$SCENE/ \
+        --data_dir $SCENE_DIR/$SCENE/colmap_0 \
+        --result_dir $RESULT_DIR/$SCENE/ \
+        --duration $NUM_FRAME \
+        --lpips_net vgg \
+        --compression stg \
+        --ckpt $RESULT_DIR/$SCENE/ckpts/ckpt_best_rank0.pt
+}
+
+GPU_LIST=(0 1 2 3 4 5)
+GPU_COUNT=${#GPU_LIST[@]}
+
+SCENE_IDX=-1
+
+for SCENE in $SCENE_LIST;
+do
+    SCENE_IDX=$((SCENE_IDX + 1))
+    {
+        run_single_scene ${GPU_LIST[$SCENE_IDX]} $SCENE
+    } &
+
+done
+
+wait
+
+# Zip the compressed files and summarize the stats
+if command -v zip &> /dev/null
+then
+    echo "Zipping results"
+    python benchmarks/stg/summarize_stats.py --results_dir $RESULT_DIR --scenes $SCENE_LIST --num_frame $NUM_FRAME
+else
+    echo "zip command not found, skipping zipping"
+fi