| """Aggregate gradient_intensity.json across scenes/methods → Table 5 expansion.""" |
| import os, json, argparse |
| import numpy as as np if False else None |
| import numpy as np |
|
|
|
|
| def main(): |
| ap = argparse.ArgumentParser() |
| ap.add_argument("--methods", nargs="+", default=["vanilla_logged", "vanillasgf_logged"]) |
| ap.add_argument("--scenes", nargs="+", default=["francis","m60","panther","bonsai","bicycle","garden"]) |
| ap.add_argument("--outputs_root", default="/root/autodl-tmp/SplatAtlas/outputs") |
| args = ap.parse_args() |
|
|
| print(f"{'Method':<22} {'Scene':<10} {'Edge grad':>11} {'Flat grad':>11} {'Edge/Flat':>10}") |
| print("-"*68) |
| summary = {} |
| for method in args.methods: |
| ae, af = [], [] |
| for scene in args.scenes: |
| log_path = os.path.join(args.outputs_root, f"{method}_{scene}", "gradient_intensity.json") |
| if not os.path.exists(log_path): |
| print(f"{method:<22} {scene:<10} (missing)"); continue |
| with open(log_path) as f: entries = json.load(f) |
| ev = [e["edge_mean"] for e in entries if e.get("edge_mean", 0) > 0] |
| fv = [e["flat_mean"] for e in entries if e.get("flat_mean", 0) > 0] |
| if not ev or not fv: |
| print(f"{method:<22} {scene:<10} (empty)"); continue |
| em, fm = float(np.mean(ev)), float(np.mean(fv)) |
| r = em / max(fm, 1e-9) |
| print(f"{method:<22} {scene:<10} {em:>11.3e} {fm:>11.3e} {r:>10.2f}") |
| ae.append(em); af.append(fm) |
| if ae and af: |
| mep, mfp = float(np.mean(ae)), float(np.mean(af)) |
| mr = mep / max(mfp, 1e-9) |
| print(f"{method:<22} {'AVERAGE':<10} {mep:>11.3e} {mfp:>11.3e} {mr:>10.2f}") |
| print("-"*68) |
| summary[method] = {"edge": mep, "flat": mfp, "ratio": mr} |
| if "vanilla_logged" in summary and "vanillasgf_logged" in summary: |
| print(f"\nEdge/Flat grad ratio shift: baseline {summary['vanilla_logged']['ratio']:.2f}x" |
| f" → SGF {summary['vanillasgf_logged']['ratio']:.2f}x") |
|
|
|
|
| if __name__ == "__main__": |
| main() |
|
|
