convert mnist to pictures in python

main.py
import numpy as np
from PIL import Image
41 chars
3 lines

main.py
with open("train-images-idx3-ubyte", "rb") as f:
    f.read(16) # skip the first 16 bytes
    data = np.fromfile(f, dtype=np.uint8).reshape(-1, 28, 28)
152 chars
4 lines

main.py
for i in range(len(data)):
    image = Image.fromarray(data[i], mode='L')
    image.save(f'images/{i}.png')
108 chars
4 lines