Keras 3 API 文档 / 内置小型数据集 / CIFAR100小图像分类数据集

CIFAR100小图像分类数据集

[source]

load_data function

keras.datasets.cifar100.load_data(label_mode="fine")

加载CIFAR100数据集.

这是一个包含50,000张32x32彩色训练图像和10,000张测试图像的数据集,标记在100个细粒度类别上,这些类别被分组为20个粗粒度类别.更多信息请参见CIFAR主页.

参数: label_mode: 可以是"fine""coarse". 如果为"fine",类别标签是细粒度标签,如果为"coarse", 输出标签是粗粒度超类.

返回: NumPy数组元组: (x_train, y_train), (x_test, y_test).

x_train: uint8 NumPy数组,包含灰度图像数据,形状为 (50000, 32, 32, 3),包含训练数据.像素值范围从0到255.

y_train: uint8 NumPy数组,包含标签(范围0-99的整数) 形状为(50000, 1),用于训练数据.

x_test: uint8 NumPy数组,包含灰度图像数据,形状为 (10000, 32, 32, 3),包含测试数据.像素值范围从0到255.

y_test: uint8 NumPy数组,包含标签(范围0-99的整数) 形状为(10000, 1),用于测试数据.

示例:

(x_train, y_train), (x_test, y_test) = keras.datasets.cifar100.load_data()
assert x_train.shape == (50000, 32, 32, 3)
assert x_test.shape == (10000, 32, 32, 3)
assert y_train.shape == (50000, 1)
assert y_test.shape == (10000, 1)