图片展示
下面是数据会集的图片展示:
class_names = [‘cats’, ‘dogs’]
plt.figure(figsize=(10, 10))
for images, labels in train_dataset.take(1):
for i in range(9):
ax = plt.subplot(3, 3, i + 1)
plt.imshow(images.numpy().astype(“uint8”))
plt.title(class_names[labels])
plt.axis(“off”)