files_to_npy.py

import sys
import numpy as np
import imageio
import glob
import os

if __name__ == "__main__":
    print("moving images in", sys.argv[1], "to", sys.argv[2])
    files = glob.glob(os.path.join(sys.argv[1], "*.png"))
    shape = imageio.imread(files[0]).shape
    data = np.zeros(shape=(len(files), *shape), dtype=np.uint8)
    for idx, f in enumerate(files):
        data[idx] = imageio.imread(f)
    np.save(sys.argv[2], data)