欢迎访问宙启技术站
智能推送

使用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数据集,并展示了一些示例图像。我们可以将这些数据用于训练和测试各种机器学习和深度学习模型。