draw image as grid on notebook

simply check below code and example result.

.

def image_grid(imgs, rows, cols):

assert len(imgs) == rows*cols

w, h = imgs[0].size

grid = Image.new('RGB', size=(cols*w, rows*h))

grid_w, grid_h = grid.size

for i, img in enumerate(imgs):

grid.paste(img, box=(i%cols*w, i//cols*h))

return grid

# read image prompt

image = Image.open("assets/images/statue.png")

depth_map = Image.open("assets/structure_controls/depth.png")

image_grid([image.resize((256, 256)), depth_map.resize((256, 256))], 1, 2)

..

Thank you.