img = make_grid(np.expand_dims(d, 1) if d.shape[0] != 3 else d)