當前位置: 首頁>>編程示例 >>用法及示例精選 >>正文


Python tf.keras.datasets.cifar10.load_data用法及代碼示例

加載 CIFAR10 數據集。

用法

tf.keras.datasets.cifar10.load_data()

返回

  • NumPy 數組元組:(x_train, y_train), (x_test, y_test)

這是一個包含 50,000 個 32x32 顏色訓練圖像和 10,000 個測試圖像的數據集,標記為 10 個類別。在 CIFAR 主頁上查看更多信息。

這些課程是:

標簽 說明
0 airplane
1 automobile
2 bird
3 cat
4 deer
5 dog
6 frog
7 horse
8 ship
9 truck

x_train:uint8 具有形狀的灰度圖像數據的 NumPy 數組(50000, 32, 32, 3),包含訓練數據。像素值範圍從 0 到 255。

y_train:uint8 NumPy 形狀的標簽數組(0-9 範圍內的整數)(50000, 1)對於訓練數據。

x_test:uint8 具有形狀的灰度圖像數據的 NumPy 數組(10000, 32, 32, 3),包含測試數據。像素值範圍從 0 到 255。

y_test:uint8 NumPy 形狀的標簽數組(0-9 範圍內的整數)(10000, 1)為測試數據。

例子:

(x_train, y_train), (x_test, y_test) = keras.datasets.cifar10.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)

相關用法


注:本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.keras.datasets.cifar10.load_data。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。