Fashion-MNIST
本数据集是每张图片大小28*28的灰度图像,共有70000张,其中包含60,000个示例的训练集和10,000个示例的测试集,10种标签
比手写体数据集更具挑战性,适合初学者(我)学习
代码实现
import tensorflow as tf
from tensorflow import keras
import numpy as np
import time
import sys
import matplotlib.pyplot as plt
from tensorflow.keras.datasets import fashion_mnist
(x_train, y_train), (x_test, y_test) = fashion_mnist.load_data()
def get_fashion_mnist_labels(labels):
text_labels = ['t-shirt', 'trouser', 'pullover', 'dress', 'coat',
'sandal', 'shirt', 'sneaker', 'bag', 'ankle boot']
return [text_labels[int(i)] for i in labels]
def show_fashion_mnist(images, labels):
_, figs = plt.subplots(1, len(images), figsize=(12, 12))
for f, img, lbl in zip(figs, images, labels):
f.imshow(img.reshape((28, 28)))
f.set_title(lbl)
f.axes.get_xaxis().set_visible(False)
f.axes.get_yaxis().set_visible(False)
plt.show()
X, y = [], []
for i in range(10):
X.append(x_train[i])
y.append(y_train[i])
show_fashion_mnist(X, get_fashion_mnist_labels(y))
batch_size = 256
if sys.platform.startswith('win'):
num_workers = 0 # 0表示不用额外的进程来加速读取数据
else:
num_workers = 4
train_iter = tf.data.Dataset.from_tensor_slices((x_train, y_train)).batch(batch_size) # 加载数据集,batch_size 小批量加载,batch_size是一个超参数
# 查看读取一遍训练数据需要的时间
start = time.time()
for X, y in train_iter:
continue
print('%.2f sec' % (time.time() - start))
输出
0.12 sec
相关注解
函数作用:不显示x、y轴坐标
f.axes.get_xaxis().set_visible(False)
f.axes.get_yaxis().set_visible(False)
tf.data.Dataset.from_tensor_slices 进行加载数据集
train_iter = tf.data.Dataset.from_tensor_slices((x_train, y_train)).batch(batch_size)