import sys
import numpy as np
from PIL import Image
n = int(sys.argv[1]); s = int(sys.argv[2]) if len(sys.argv) > 2 else 2
e = np.load("npy/identity_N%d.npy" % n)
pal = np.array([255, 190, 110, 20], dtype=np.uint8)
img = np.kron(pal[e], np.ones((s, s), dtype=np.uint8))
Image.fromarray(img, "L").save("identity_N%d.png" % n)
print("identity_N%d.png" % n, img.shape)