paddle.vision / datasets / MNIST

MNIST¶

class paddle.vision. MNIST [源代码] ¶: MNIST 数据集

参数¶

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

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

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

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

download (bool) - 是否自定下载数据集文件。默认为 True 。

返回¶

MNIST数据集实例

代码示例¶

from paddle.vision.datasets import MNIST

mnist = MNIST(mode='test')

for i in range(len(mnist)):
    sample = mnist[i]
    print(sample[0].shape, sample[1])

此页内容是否对您有帮助

PaddlePaddle

paddle

paddle.amp

paddle.callbacks

paddle.compat

paddle.distributed

paddle.distribution

paddle.fluid

paddle.io

paddle.jit

paddle.metric

paddle.nn

paddle.onnx

paddle.optimizer

paddle.regularizer

paddle.static

paddle.sysconfig

paddle.text

paddle.utils

paddle.vision

paddle.vision / datasets / MNIST

MNIST¶

参数¶

返回¶

代码示例¶