File size: 750 Bytes
986404c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | from pathlib import Path
import argparse
import numpy as np
def main():
parser = argparse.ArgumentParser()
parser.add_argument("--output", default="data/fake_corrdiff.npz")
parser.add_argument("--samples", type=int, default=1)
args = parser.parse_args()
path = Path(args.output)
path.parent.mkdir(parents=True, exist_ok=True)
rng = np.random.default_rng(42)
inputs = rng.standard_normal((args.samples, 12, 36, 36)).astype("float32")
targets = rng.standard_normal((args.samples, 4, 448, 448)).astype("float32")
np.savez_compressed(path, input=inputs, target=targets)
print(f"input: {inputs.shape}")
print(f"target: {targets.shape}")
print(f"saved: {path}")
if __name__ == "__main__":
main()
|