Fastfill-test / code /fastfill /adapters /internscenes_arkit.py
XXXpilar's picture
FastFill v4 base (step 6552): LoRA adapter, geometry heads, tokenizer, training code, predict.py
e6f80f3 verified
Raw History Blame Contribute Delete
786 Bytes
"""InternScenes Real2Sim re-layout of ARKitScenes videos (conventions: see internscenes.py).
Group = arkit:<video_id>; several videos of one visit (same venue) are listed in meta.group_aliases."""
from fastfill.adapters.internscenes import arkit_visits, load_real
SOURCE = "InternScenes_arkit"
def load(root):
visit = arkit_visits(root)
yield from load_real(root, "arkitscenes_exported.json", SOURCE,
lambda s: "arkit:" + s["source_scene_name"],
lambda s: {"video_id": s["source_scene_name"], "visit_id": visit.get(s["source_scene_name"]),
"group_aliases": [f"arkit_visit:{visit[s['source_scene_name']]}"]
if visit.get(s["source_scene_name"]) else []})