| """Aggregate densification_events.json across scenes/methods, emit Table 8 data.""" |
| import os |
| import json |
| import argparse |
|
|
|
|
| def main(): |
| ap = argparse.ArgumentParser() |
| ap.add_argument("--scenes", nargs="+", |
| default=["francis", "m60", "panther", "bonsai", "bicycle"]) |
| ap.add_argument("--methods", nargs="+", |
| default=["vanilla_logged", "vanillasgf_logged"]) |
| ap.add_argument("--outputs_root", default="/root/autodl-tmp/SplatAtlas/outputs") |
| args = ap.parse_args() |
|
|
| print(f"{'Method':<22} {'Scene':<10} {'#Events':>9} {'#Edge':>8} {'#Flat':>8} {'Edge/Flat':>10}") |
| print("-" * 72) |
|
|
| summary = {} |
| for method in args.methods: |
| agg_total = agg_edge = agg_flat = 0 |
| for scene in args.scenes: |
| log_path = os.path.join(args.outputs_root, f"{method}_{scene}", |
| "densification_events.json") |
| if not os.path.exists(log_path): |
| print(f"{method:<22} {scene:<10} {'(missing)':>9}") |
| continue |
| with open(log_path) as f: |
| events = json.load(f) |
| total = sum(e["n_will_densify"] for e in events) |
| edge = sum(e["n_edge"] for e in events) |
| flat = sum(e["n_flat"] for e in events) |
| ratio = edge / max(flat, 1) |
| agg_total += total; agg_edge += edge; agg_flat += flat |
| print(f"{method:<22} {scene:<10} {total:>9} {edge:>8} {flat:>8} {ratio:>10.2f}") |
| m_ratio = agg_edge / max(agg_flat, 1) |
| print(f"{method:<22} {'AVERAGE':<10} {agg_total:>9} {agg_edge:>8} {agg_flat:>8} {m_ratio:>10.2f}") |
| print("-" * 72) |
| summary[method] = {"total": agg_total, "edge": agg_edge, "flat": agg_flat, "ratio": m_ratio} |
|
|
| if "vanilla_logged" in summary and "vanillasgf_logged" in summary: |
| base_r = summary["vanilla_logged"]["ratio"] |
| sgf_r = summary["vanillasgf_logged"]["ratio"] |
| print(f"\nEdge/Flat ratio shift: baseline {base_r:.2f}x → SGF {sgf_r:.2f}x" |
| f" (Δ = {sgf_r - base_r:+.2f})") |
|
|
|
|
| if __name__ == "__main__": |
| main() |
|
|
