Function save_image_grid

eval_code/recons/utils/vis_utils.py:9–22

images: numpy array of shape (N, H, W, 3) grid_shape: (rows, cols)

(images: np.ndarray, grid_shape: tuple, save_path: str)

def save_image_grid(images: np.ndarray, grid_shape: tuple, save_path: str):
    """
    images: numpy array of shape (N, H, W, 3)
    grid_shape: (rows, cols)
    """
    H, W = images.shape[1], images.shape[2]
    grid = np.zeros((grid_shape[0]*H, grid_shape[1]*W, 3), dtype=np.uint8)
    
    for i in range(min(len(images), grid_shape[0]*grid_shape[1])):
        row = i // grid_shape[1]
        col = i % grid_shape[1]
        grid[row*H:(row+1)*H, col*W:(col+1)*W] = images[i]
    
    Image.fromarray(grid).save(save_path)


def save_image_grid_auto(images: Union[np.ndarray, torch.Tensor], save_path: str):