MindSpore加载图像数据集

 时间:2026-04-24 04:44:16

1、一、准备数据集

下载MNIST数据集的训练图像和标签并解压,存放在./MNIST路径中,目录结构如下。

导入mindspore.dataset模块。

import mindspore.dataset as ds

MindSpore加载图像数据集

2、二、加载数据集

MindSpore目前支持加载图像领域常用的经典数据集和多种数据存储格式下的数据集,用户也可以通过构建自定义数据集类实现自定义方式的数据加载。

下面演示使用mindspore.dataset模块中的MnistDataset类加载MNIST数据集。

3、1. 配置数据集目录,创建MNIST数据集对象。

DATA_DIR = "./MNIST"

mnist_dataset = ds.MnistDataset(DATA_DIR, num_samples=6, shuffle=False)

4、2. 创建迭代器,通过迭代器获取数据。

import matplotlib.pyplot as plt

mnist_it = mnist_dataset.create_dict_iterator()

data = mnist_it.get_next()

plt.imshow(data['image'].asnumpy().squeeze(), cmap=plt.cm.gray)

plt.title(data['label'].asnumpy(), fontsize=20)

plt.show()

图片展示如下:

MindSpore加载图像数据集

5、此外,用户还可以在数据集加载时传入sampler指定数据采样方式。MindSpore目前支持的数据采样器及其详细使用方法,可参考编程指南中采样器章节。

https://www.mindspore.cn/doc/programming_guide/zh-CN/master/sampler.html

  • MindSpore的MNIST图片分类如何在jupyter中运行
  • 里讯浏览器怎么关闭加载图像
  • windows10环境下安装mindspore经验
  • 机器学习之建模,导入数据集,划分数据集
  • MindSpore如何动态调整学习率
  • 热门搜索
    编发图片大全 一千部禁片免费观看大全 向前冲舞蹈视频大全 冰淇淋图片大全 乐谱大全 经典儿歌大全100首 万家乐壁挂炉怎么样 捷达车怎么样 米芾简介 孕妇脚气怎么办