Function grayscale_grid_vis

lib/vis.py:4–13

(X, (nh, nw), save_path=None)

from scipy.misc import imsave

def grayscale_grid_vis(X, (nh, nw), save_path=None):
    h, w = X[0].shape[:2]
    img = np.zeros((h*nh, w*nw))
    for n, x in enumerate(X):
        j = n/nw
        i = n%nw
        img[j*h:j*h+h, i*w:i*w+w] = x
    if save_path is not None:
        imsave(save_path, img)
    return img

def color_grid_vis(X, (nh, nw), save_path=None):
    h, w = X[0].shape[:2]