MNIST

class paddle.vision.datasets. MNIST [源代码]

MNIST 数据集

参数

  • image_path (str) - 图像文件路径,如果 download 参数设置为 Trueimage_path 参数可以设置为 None 。默认值为 None

  • label_path (str) - 标签文件路径,如果 download 参数设置为 Truelabel_path 参数可以设置为 None 。默认值为 None

  • chw_format (bool) - 若为 True 输出形状为 [1, 28, 28] , 否则为 [1, 784] 。默认值为 True

  • mode (str) - 'train''test' 模式,默认为 'train'

  • download (bool) - 当 data_fileNone 时,该参数决定是否自动下载数据集文件。默认为 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])