MNIST

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

MNIST 数据集

参数

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

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

  • mode (str) - 'train' 或 'test' 模式，默认为 'train' 。

  • download (bool) - 当 data_file 是 None 时，该参数决定是否自动下载数据集文件。默认为 True 。

  • backend(str，可选）- 指定图像的返回类型：PIL.Image 或 numpy.ndarray。应该是{'pil'，'cv2'} 之一。如果未设置此选项，将从paddle.vsion.get_image_backend获取，默认为 pil 。 默认值：None。

返回

MNIST数据集实例

代码示例

  1. from paddle.vision.datasets import MNIST
  3. mnist = MNIST(mode='test')
  5. for i in range(len(mnist)):
  6.     sample = mnist[i]
  7.     print(sample[0].size, sample[1])

使用本API的教程文档