File size: 1,122 Bytes
5c365c5 | 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 | """Generate an NPZ fixture with the ACE 40/44 channel contract."""
from __future__ import annotations
import argparse
import sys
from pathlib import Path
if __package__ in (None, ""):
sys.path.insert(0, str(Path(__file__).resolve().parents[2]))
from ACE.model.data import save_fake_pairs
from ACE.model.paths import GENERATED_DATA_PATH
def main() -> int:
parser = argparse.ArgumentParser(description=__doc__)
parser.add_argument(
"--output-path", type=Path, default=GENERATED_DATA_PATH,
help="Output NPZ path (default: ACE/data/data/ace_pairs.npz)",
)
parser.add_argument("--num-samples", type=int, default=8)
parser.add_argument("--height", type=int, default=180)
parser.add_argument("--width", type=int, default=360)
parser.add_argument("--seed", type=int, default=0)
args = parser.parse_args()
output = save_fake_pairs(
args.output_path,
num_samples=args.num_samples,
height=args.height,
width=args.width,
seed=args.seed,
)
print(output)
return 0
if __name__ == "__main__":
raise SystemExit(main())
|