SplatAtlas / scripts /run_all_benchmarks_2.sh
KCBtheone's picture
Upload SplatAtlas benchmark pipeline code
23e73f9 verified
Raw
History Blame Contribute Delete
2.56 kB
#!/bin/bash
# 初始化 Conda 环境路径
CONDA_BASE=$(conda info --base)
source "$CONDA_BASE/etc/profile.d/conda.sh"
# 确保主路径在 PYTHONPATH 中
export PYTHONPATH=$PYTHONPATH:/root/autodl-tmp/SplatAtlas
# 定义任务队列:(环境绝对路径 脚本相对路径)
declare -a tasks=(
"/root/autodl-tmp/envs/gaussianpro scripts/run_gaussianpro_benchmark.py"
"/root/autodl-tmp/envs/conegs_env scripts/run_conegs_benchmark.py"
"/root/autodl-tmp/envs/steepgs_env scripts/run_steepgs.py"
"/root/autodl-tmp/envs/absgs_env scripts/run_absgs_benchmark.py"
"/root/autodl-tmp/envs/reactgs_env scripts/run_reactgs_benchmark.py"
"/root/autodl-tmp/envs/lapis_gs scripts/run_lapisgs_benchmark.py"
"/root/autodl-tmp/envs/co_adaptation_3dgs scripts/run_coadaptgs_benchmark.py"
"/root/autodl-tmp/envs/mini_splatting scripts/run_minisplatting_benchmark.py"
"/root/autodl-tmp/envs/gs_lpm scripts/run_gslpm.py"
"/root/autodl-tmp/envs/opti3dgs scripts/run_opti3dgs_benchmark.py"
"/root/autodl-tmp/envs/ghap_env scripts/run_ghap_benchmark.py"
"/root/autodl-tmp/envs/3dgs_mcmc scripts/run_3dgsmcmc_benchmark.py"
# "/root/autodl-tmp/envs/lod_gs scripts/run_lod_gs_benchmark.py"
#conegs可能需要重跑,test可能没有被划分
)
echo "🚀 SplatAtlas: Starting Robust Optimization Pipeline..."
for task in "${tasks[@]}"; do
# 解析环境路径和脚本
read -r env_path script_path <<< "$task"
echo "----------------------------------------------------------------"
echo "🔄 Method: $(basename "$script_path")"
# 1. 激活环境
conda activate "$env_path"
if [ $? -eq 0 ]; then
# 2. 环境自动体检:添加了 scikit-image 的安装
echo "🛠️ Checking requirements in $env_path..."
# 增加 scikit-image 以解决 ModuleNotFoundError: No module named 'skimage'
pip install lpips imageio scikit-image -q
# 3. 执行 Benchmark
echo "✅ Environment ready. Executing..."
"$CONDA_PREFIX/bin/python" "/root/autodl-tmp/SplatAtlas/$script_path"
if [ $? -eq 0 ]; then
echo "✨ Finished: $(basename "$script_path")"
else
echo "❌ Error: $(basename "$script_path") failed."
fi
else
echo "❌ Failed to activate: $env_path"
fi
# 释放当前环境
conda deactivate
done
echo "================================================================"
echo "🏁 Optimization Pipeline completed."