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数据集实例。
代码示例¶
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])