KCBtheone's picture
Upload SplatAtlas benchmark pipeline code
23e73f9 verified
Raw
History Blame Contribute Delete
746 Bytes
import subprocess
s = "lighthouse"
print(f"\n>> Testing vanilla_3dgs on {s} (EXIF/Images2 Match)...")
cmd = [
"python", "scripts/phase1_validation/render_single.py",
"--ply_path", f"/root/autodl-tmp/SplatAtlas/outputs/vanilla_3dgs_{s}/point_cloud/iteration_30000/point_cloud.ply",
"--source_path", f"/root/autodl-tmp/dataset/tnt/{s}",
"--model_path", f"/root/autodl-tmp/SplatAtlas/outputs/vanilla_3dgs_{s}",
"--output_dir", f"/tmp/sanity_tnt_{s}",
"--resolution", "2",
"--bg_color", "0,0,0"
]
res = subprocess.run(cmd, capture_output=True, text=True, cwd="/root/autodl-tmp/SplatAtlas")
for line in res.stdout.split("\n"):
if "Mean PSNR" in line or "Delta" in line or "STATUS" in line:
print(f" {line}")