MindSpore加载图像数据集

 时间:2026-02-20 07:40:09

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如何动态调整学习率
  • 热门搜索
    a字裙搭配什么上衣 arms是什么意思 淼淼是什么意思 画电路图用什么软件 3166什么意思 信用卡是什么意思 ufo是什么意思 女生说吃蘑菇什么意思 蔺相如是个什么样的人 famous是什么意思