File size: 2,075 Bytes
cca6827 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 | #!/usr/bin/env python3
"""Build the portable held-out (NV) view indices used by eval_final.py (FB150 + Omni300 only;
Toys4K-300 keeps the original evaluate_appforce protocol and has no held-out index).
Input: the absolute-path indices written by heldout/prepare_heldout.py
(heldout_fb150.json, heldout_omni300.json).
Output: heldout_index/{fb150,omni300}.json =
{object: [{tag, cam, px, src: "exp"|"heldout", kind: "heldout"|"saved"|"pool", vis?}, ...]}
src=exp -> files live in the dataset exp dir (inputs/<obj>_<tag>.png, renders/<obj>_<tag>.npz)
src=heldout -> files live in the held-out root (data/fb150_heldout.tar, omni pool)
No record is filtered here; the visibility threshold (--min-vis) is applied by eval_final.py.
"""
import argparse, json
from pathlib import Path
LP = "/lp-dev/jonghoon/mv-mesh"
def main():
ap = argparse.ArgumentParser()
ap.add_argument("--src-dir", type=Path, default=Path(f"{LP}/.debug/eval_heldout_nv"))
ap.add_argument("--out", type=Path, default=Path(__file__).resolve().parent / "heldout_index")
a = ap.parse_args()
a.out.mkdir(parents=True, exist_ok=True)
fb = json.loads((a.src_dir / "heldout_fb150.json").read_text())
fb2 = {o: [dict(tag=r["tag"], cam=r["cam"], vis=r["vis"], px=r["px"], src="heldout", kind="heldout") for r in v]
for o, v in fb.items()}
om = json.loads((a.src_dir / "heldout_omni300.json").read_text())
om2 = {o: [dict(tag=r["tag"], cam=r["cam"], px=r["px"], src="exp" if r["kind"] == "saved" else "heldout",
kind=r["kind"]) for r in v] for o, v in om.items()}
for name, d in (("fb150", fb2), ("omni300", om2)):
(a.out / f"{name}.json").write_text(json.dumps(d, separators=(",", ":")))
n = [len(v) for v in d.values()]
print(name, "objects", len(d), "records", sum(n), "min/obj", min(n))
fbv = [len([r for r in v if r["vis"] >= 0.2]) for v in fb2.values()]
print("fb150 vis>=0.20: records", sum(fbv), "min/obj", min(fbv), "objs with 0:", sum(x == 0 for x in fbv))
if __name__ == "__main__":
main()
|