peal.visualization.image_grid.make_column¶

peal.visualization.image_grid.make_column(images, labels, title)[source]¶

Stack images vertically into one titled, captioned column.

Parameters:

images (torch.Tensor) – [N, 3, H, W] floats in [0, 1] on the CPU.
labels (list of str) – Caption drawn under each image; N entries.
title (str) – Column title, rendered rotated by 90 degrees above the first image.

Returns:

uint8 RGB array of shape [(N + 2) * (H + H // 2) + H // 2, W + 2 * (H // 8), 3] with a white background. The height depends only on N and H, which is what lets make_image_grid() concatenate columns.

Return type:

numpy.ndarray