第1关:数据集加载
任务描述
MindSpore 可以帮助你加载常见的数据集、特定数据格式的数据集或自定义的数据集。加载数据集时,需先导入所需要依赖的库 mindspore.dataset。
本关任务:熟悉使用 MindSpore 框架加载数据集。
import mindspore.dataset as ds
import numpy as np
def get():
# 常见数据集的加载
DATA_DIR = "./MindSpore/src/step1/cifar-10-binary/cifar-10-batches-bin"
# ********** Begin *********#
# 根据数据地址初始化数据集 cifar10_dataset
cifar10_dataset = ds.Cifar10Dataset(DATA_DIR)
# ********** End **********#
for data in cifar10_dataset.create_dict_iterator():
# In CIFAR-10 dataset, each dictionary of data has keys "image" and "label".
# 图像数据比较大,平台不方便展示
#print(data["image"])
print(data[