使用Keras.datasets.cifar10库中的load_data()函数加载CIFAR-10数据集
发布时间:2023-12-25 21:01:27
CIFAR-10是一个包含60,000个32x32大小的彩色图像的数据集,其中包含10个不同的类别。Keras.datasets.cifar10库提供了一个方便的方法来加载和使用这个数据集。
加载CIFAR-10数据集的 步是导入必要的库,并调用load_data()函数。
以下是一个使用例子:
from keras.datasets import cifar10 # 加载CIFAR-10数据集 (x_train, y_train), (x_test, y_test) = cifar10.load_data()
在上面的例子中,我们使用load_data()函数分别加载了训练集和测试集。该函数将返回一个元组,其中包含图像数据及其对应的标签。
训练集包含50,000个图像,测试集包含10,000个图像。每个图像都是32x32的彩色图像,像素值范围在0到255之间。标签是一个0到9之间的整数,表示图像所属的类别。
让我们看一下加载的数据集的一些示例:
import matplotlib.pyplot as plt
# 显示数据集的一些图像
fig, axes = plt.subplots(2, 5, figsize=(10, 5))
axes = axes.ravel()
for i in range(10):
axes[i].imshow(x_train[i])
axes[i].set_title(y_train[i])
axes[i].axis('off')
plt.tight_layout()
plt.show()
在上面的代码中,我们使用Matplotlib库来显示数据集的一些图像。我们创建一个包含2行5列的子图,然后将加载的 个10个训练图像显示在这些子图中。
请注意,在使用imshow()函数显示图像之前,我们使用set_title()和axis('off')函数添加图像的标题,并关闭坐标轴。
运行上面的代码,将会生成一个包含10个图像的子图。每个图像的标题将是其对应的标签,这个标签将指示图像所属的类别。
现在,我们已经成功加载了CIFAR-10数据集,并展示了一些示例图像。我们可以将这些数据用于训练和测试各种机器学习和深度学习模型。
