Python matplotlib: 使用 plt.figure() 和 plt.subplot() 显示训练数据图像

这段代码使用 Python 的 matplotlib 库来显示训练数据集中的一些图像及其对应的标签。

• plt.figure(figsize=(10, 10)) 用于创建一个 10x10 英寸大小的图形窗口。
• train_ds.take(1) 从训练数据集中获取一个批次的数据，包含多个图像和对应的标签。
• for images, labels in train_ds.take(1): 循环遍历这个批次中的每个图像和标签。
• plt.subplot(5, 5, i + 1) 在窗口中创建一个 5x5 的子图网格，并选择第 (i + 1) 个子图。
• plt.imshow(images[i].numpy().astype('uint8')) 在当前子图中显示第 i 个图像。
• plt.title(class_names[labels[i]]) 为当前图像设置标题，标题内容是对应标签的名称。
• plt.axis('off') 关闭当前子图的坐标轴。

最终，这段代码会在一个窗口中显示 5x5 共 25 个图像和它们的标签。