25 lines
644 B
Python
25 lines
644 B
Python
import matplotlib.pyplot as plt
|
|
from typing import List
|
|
from PIL.Image import Image
|
|
from math import ceil
|
|
|
|
|
|
def display_images(images: List[Image], titles: List[str], images_per_row: int = 3):
|
|
fig, axes = plt.subplots(
|
|
nrows=ceil(len(images) / images_per_row),
|
|
ncols=min(images_per_row, len(images)),
|
|
figsize=(12, 8),
|
|
)
|
|
|
|
axes = axes.flatten()
|
|
|
|
for i, (title, image) in enumerate(zip(titles, images)):
|
|
axes[i].imshow(image)
|
|
axes[i].axis("off")
|
|
axes[i].set_title(title)
|
|
|
|
for i in range(len(images), len(axes)):
|
|
axes[i].axis("off")
|
|
|
|
plt.tight_layout()
|
|
plt.show()
|