CIFAR-10數據集下載地址

'http://www.cs.toronto.edu/~kriz/cifar-10-binary.tar.gz'

下載TensorFlow Models代碼

https://github.com/tensorflow/models.git

拷貝cifar10.py和cifar10_input.py到當前目錄

編寫代碼

import cifar10 import cifar10_input import tensorflow as tf #下載和解壓縮 cifar10.maybe_download_and_extract()data_dir = '/tmp/cifar10_data/cifar-10-batches-bin' batch_size = 100#生成CIFAR-10的訓練數據和訓練標簽數據 images_train, labels_train = cifar10_input.distorted_inputs(data_dir=data_dir, batch_size=batch_size)#生成CIFAR-10的測試數據和測試標簽數據 images_test, labels_test = cifar10_input.inputs(eval_data=True, data_dir=data_dir, batch_size=batch_size)sess = tf.InteractiveSession() tf.global_variables_initializer() tf.train.start_queue_runners() print(images_train) print(images_test)

總結

以上是生活随笔為你收集整理的TensorFlow CIFAR-10数据集的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。