import matplotlib.pyplot as plt

print("可视化花卉数据集")
plt.figure(figsize=(10, 10))
for images, labels in train_ds.take(1):
   for i in range(6):
      ax = plt.subplot(3, 3, i + 1)
      plt.imshow(images[i].numpy().astype("uint8"))
      plt.title(class_names[labels[i]])
      plt.axis("off")

print("迭代数据集")
print("检索图像的批次")
for image_batch, labels_batch in train_ds:
   print(image_batch.shape)
   print(labels_batch.shape)
   break

可视化花卉数据集
迭代数据集
检索图像的批次
(32, 180, 180, 3)
(32,)