MNIST¶
参数¶
image_path (str) - 图像文件路径,如果
download
参数设置为True
,image_path
参数可以设置为None
。默认值为None
。label_path (str) - 标签文件路径,如果
download
参数设置为True
,label_path
参数可以设置为None
。默认值为None
。chw_format (bool) - 若为
True
输出形状为[1, 28, 28]
, 否则为[1, 784]
。默认值为True
。mode (str) -
'train'
或'test'
模式,默认为'train'
。download (bool) - 当
data_file
是None
时,该参数决定是否自动下载数据集文件。默认为True
。
返回¶
MNIST数据集实例
代码示例¶
from paddle.vision.datasets import MNIST mnist = MNIST(mode='test') for i in range(len(mnist)): sample = mnist[i] print(sample[0].size, sample[1])