import numpy as np
import matplotlib.pyplot as plt
from tensorflow.keras.datasets import cifar10
# 加载数据集(首次运行会自动下载)
(train_images, train_labels), (_, _) = cifar10.load_data()
# 查看第一张图片
image = train_images[0] # 形状为(32, 32, 3)
label = train_labels[0][0] # 标签编号
# 显示图片
plt.imshow(image)
plt.title(f"Label: {label}")
plt.axis('off')
plt.show()