| 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) | |