get_sample_grid.py

import random 
from PIL import Image
import glob
import numpy as np 

def get_sample_grid_(paths, grid_w=4, grid_h=4, imsize_w=256, imsize_h=256):
    imgs = []
    for p in paths:
        imgs.append(np.array(Image.open(p).convert('RGB').resize([imsize_w,imsize_h])))


    rows = []
    for i in range(grid_h):
        row = np.hstack(imgs[i*grid_w:(i+1)*grid_w])
        rows.append(row)

    return Image.fromarray(np.vstack(rows))

def get_sample_grid(dir_path, grid_w=4, grid_h=4, imsize_w=256, imsize_h=256):
    
    paths = glob.glob(dir_path)

    samples = random.sample(paths, grid_w * grid_h)

    return get_sample_grid_(samples, grid_w, grid_h, imsize_w, imsize_h)