| |
| """Pre-generate + cache synthetic scenes (renders GT with the rasterizer). |
| |
| python scripts/gen_synthetic.py --config configs/synthetic_smoke.yaml --split train --n-scenes 64 |
| """ |
|
|
| import argparse |
|
|
| from mapgs.config import load_config |
| from mapgs.data import SyntheticDataset |
|
|
|
|
| def main(): |
| ap = argparse.ArgumentParser() |
| ap.add_argument("--config", type=str, default="configs/synthetic_smoke.yaml") |
| ap.add_argument("--override", nargs="*", default=[]) |
| ap.add_argument("--split", type=str, default="train") |
| ap.add_argument("--n-scenes", type=int, default=None) |
| ap.add_argument("--regenerate", action="store_true") |
| args = ap.parse_args() |
|
|
| cfg = load_config(args.config, args.override) |
| ds = SyntheticDataset(cfg, split=args.split, n_scenes=args.n_scenes, |
| device=cfg.device, regenerate=args.regenerate) |
| print(f"prepared {len(ds)} {args.split} scenes under {ds.root}") |
|
|
|
|
| if __name__ == "__main__": |
| main() |
|
|
