| """Create realistic packed-wedge synthetic data when licensed observations are unavailable.""" |
|
|
| from __future__ import annotations |
|
|
| import argparse |
| from pathlib import Path |
|
|
| import numpy as np |
| import sys |
| ROOT=Path(__file__).resolve().parents[1];sys.path.insert(0,str(ROOT)) |
|
|
| from model.echocast_3d import load_config, make_sample |
|
|
|
|
| def main() -> None: |
| parser = argparse.ArgumentParser() |
| parser.add_argument("--config", default="conf/config.yaml") |
| parser.add_argument("--output", default="data") |
| args = parser.parse_args() |
| config = load_config(ROOT / args.config) |
| output = ROOT / args.output |
| output.mkdir(parents=True, exist_ok=True) |
| for split, count in (("train", config["data"]["train_samples"]), ("validation", config["data"]["validation_samples"])): |
| for index in range(count): |
| sample = make_sample(config, config["seed"] + index + (10000 if split == "validation" else 0)) |
| path = output / f"{split}_{index:04d}.npz" |
| np.savez(path, **sample) |
| print(f"wrote {path} ({path.stat().st_size / 1024**2:.2f} MiB)") |
|
|
|
|
| if __name__ == "__main__": |
| main() |
|
|
