2021-11-29 11:36:30 +08:00
|
|
|
|
mindspore.dataset
|
|
|
|
|
|
=================
|
|
|
|
|
|
|
2021-12-24 11:35:46 +08:00
|
|
|
|
该模块提供了加载和处理各种通用数据集的API,如MNIST、CIFAR-10、CIFAR-100、VOC、COCO、ImageNet、CelebA、CLUE等,
|
|
|
|
|
|
也支持加载业界标准格式的数据集,包括MindRecord、TFRecord、Manifest等。此外,用户还可以使用此模块定义和加载自己的数据集。
|
|
|
|
|
|
|
|
|
|
|
|
该模块还提供了在加载时进行数据采样的API,如SequentialSample、RandomSampler、DistributedSampler等。
|
|
|
|
|
|
|
|
|
|
|
|
大多数数据集可以通过指定参数 `cache` 启用缓存服务,以提升整体数据处理效率。
|
|
|
|
|
|
请注意Windows平台上还不支持缓存服务,因此在Windows上加载和处理数据时,请勿使用。更多介绍和限制,
|
2022-04-12 15:45:19 +08:00
|
|
|
|
请参考 `Single-Node Tensor Cache <https://www.mindspore.cn/tutorials/experts/zh-CN/master/dataset/cache.html>`_。
|
2021-12-24 11:35:46 +08:00
|
|
|
|
|
|
|
|
|
|
在API示例中,常用的模块导入方法如下:
|
|
|
|
|
|
|
|
|
|
|
|
.. code-block::
|
|
|
|
|
|
|
|
|
|
|
|
import mindspore.dataset as ds
|
|
|
|
|
|
from mindspore.dataset.transforms import c_transforms
|
|
|
|
|
|
|
2022-02-08 15:25:43 +08:00
|
|
|
|
常用数据集术语说明如下:
|
|
|
|
|
|
|
|
|
|
|
|
- Dataset,所有数据集的基类,提供了数据处理方法来帮助预处理数据。
|
|
|
|
|
|
- SourceDataset,一个抽象类,表示数据集管道的来源,从文件和数据库等数据源生成数据。
|
|
|
|
|
|
- MappableDataset,一个抽象类,表示支持随机访问的源数据集。
|
|
|
|
|
|
- Iterator,用于枚举元素的数据集迭代器的基类。
|
|
|
|
|
|
|
2022-02-11 09:07:57 +08:00
|
|
|
|
视觉
|
|
|
|
|
|
-----
|
2021-11-29 11:36:30 +08:00
|
|
|
|
|
2022-02-15 16:22:46 +08:00
|
|
|
|
.. mscnautosummary::
|
2021-11-29 11:36:30 +08:00
|
|
|
|
:toctree: dataset
|
2022-02-11 17:34:24 +08:00
|
|
|
|
:nosignatures:
|
|
|
|
|
|
:template: classtemplate_inherited.rst
|
2021-11-29 11:36:30 +08:00
|
|
|
|
|
2022-02-11 17:34:24 +08:00
|
|
|
|
mindspore.dataset.Caltech101Dataset
|
|
|
|
|
|
mindspore.dataset.Caltech256Dataset
|
2021-12-04 11:21:09 +08:00
|
|
|
|
mindspore.dataset.CelebADataset
|
2022-02-11 15:23:20 +08:00
|
|
|
|
mindspore.dataset.Cifar10Dataset
|
|
|
|
|
|
mindspore.dataset.Cifar100Dataset
|
2022-02-11 17:34:24 +08:00
|
|
|
|
mindspore.dataset.CityscapesDataset
|
2021-12-04 11:21:09 +08:00
|
|
|
|
mindspore.dataset.CocoDataset
|
2022-02-11 17:34:24 +08:00
|
|
|
|
mindspore.dataset.DIV2KDataset
|
|
|
|
|
|
mindspore.dataset.EMnistDataset
|
|
|
|
|
|
mindspore.dataset.FakeImageDataset
|
|
|
|
|
|
mindspore.dataset.FashionMnistDataset
|
|
|
|
|
|
mindspore.dataset.FlickrDataset
|
|
|
|
|
|
mindspore.dataset.Flowers102Dataset
|
2021-11-29 11:36:30 +08:00
|
|
|
|
mindspore.dataset.ImageFolderDataset
|
2022-02-11 17:34:24 +08:00
|
|
|
|
mindspore.dataset.KMnistDataset
|
2022-02-11 15:23:20 +08:00
|
|
|
|
mindspore.dataset.ManifestDataset
|
2021-11-29 11:36:30 +08:00
|
|
|
|
mindspore.dataset.MnistDataset
|
2022-02-11 17:34:24 +08:00
|
|
|
|
mindspore.dataset.PhotoTourDataset
|
|
|
|
|
|
mindspore.dataset.Places365Dataset
|
|
|
|
|
|
mindspore.dataset.QMnistDataset
|
|
|
|
|
|
mindspore.dataset.SBDataset
|
|
|
|
|
|
mindspore.dataset.SBUDataset
|
|
|
|
|
|
mindspore.dataset.SemeionDataset
|
|
|
|
|
|
mindspore.dataset.STL10Dataset
|
|
|
|
|
|
mindspore.dataset.SVHNDataset
|
|
|
|
|
|
mindspore.dataset.USPSDataset
|
2021-11-29 11:36:30 +08:00
|
|
|
|
mindspore.dataset.VOCDataset
|
2022-02-11 17:34:24 +08:00
|
|
|
|
mindspore.dataset.WIDERFaceDataset
|
2021-11-29 11:36:30 +08:00
|
|
|
|
|
2022-02-11 09:07:57 +08:00
|
|
|
|
文本
|
2021-11-29 11:36:30 +08:00
|
|
|
|
----
|
|
|
|
|
|
|
2022-02-15 16:22:46 +08:00
|
|
|
|
.. mscnautosummary::
|
2021-11-29 11:36:30 +08:00
|
|
|
|
:toctree: dataset
|
2022-02-11 17:34:24 +08:00
|
|
|
|
:nosignatures:
|
|
|
|
|
|
:template: classtemplate_inherited.rst
|
2021-11-29 11:36:30 +08:00
|
|
|
|
|
2022-02-11 17:34:24 +08:00
|
|
|
|
mindspore.dataset.AGNewsDataset
|
|
|
|
|
|
mindspore.dataset.AmazonReviewDataset
|
2021-11-29 11:36:30 +08:00
|
|
|
|
mindspore.dataset.CLUEDataset
|
2022-02-11 17:34:24 +08:00
|
|
|
|
mindspore.dataset.CoNLL2000Dataset
|
|
|
|
|
|
mindspore.dataset.DBpediaDataset
|
|
|
|
|
|
mindspore.dataset.EnWik9Dataset
|
|
|
|
|
|
mindspore.dataset.IMDBDataset
|
|
|
|
|
|
mindspore.dataset.IWSLT2016Dataset
|
|
|
|
|
|
mindspore.dataset.IWSLT2017Dataset
|
|
|
|
|
|
mindspore.dataset.PennTreebankDataset
|
|
|
|
|
|
mindspore.dataset.SogouNewsDataset
|
2022-02-11 15:23:20 +08:00
|
|
|
|
mindspore.dataset.TextFileDataset
|
2022-02-11 17:34:24 +08:00
|
|
|
|
mindspore.dataset.UDPOSDataset
|
|
|
|
|
|
mindspore.dataset.WikiTextDataset
|
|
|
|
|
|
mindspore.dataset.YahooAnswersDataset
|
|
|
|
|
|
mindspore.dataset.YelpReviewDataset
|
|
|
|
|
|
|
|
|
|
|
|
音频
|
|
|
|
|
|
------
|
|
|
|
|
|
|
2022-02-15 16:22:46 +08:00
|
|
|
|
.. mscnautosummary::
|
2022-02-11 17:34:24 +08:00
|
|
|
|
:toctree: dataset
|
|
|
|
|
|
:nosignatures:
|
|
|
|
|
|
:template: classtemplate_inherited.rst
|
|
|
|
|
|
|
|
|
|
|
|
mindspore.dataset.LJSpeechDataset
|
|
|
|
|
|
mindspore.dataset.SpeechCommandsDataset
|
|
|
|
|
|
mindspore.dataset.TedliumDataset
|
|
|
|
|
|
mindspore.dataset.YesNoDataset
|
2021-11-29 11:36:30 +08:00
|
|
|
|
|
2022-02-11 15:23:20 +08:00
|
|
|
|
标准格式
|
|
|
|
|
|
--------
|
2021-11-29 11:36:30 +08:00
|
|
|
|
|
2022-02-15 16:22:46 +08:00
|
|
|
|
.. mscnautosummary::
|
2021-11-29 11:36:30 +08:00
|
|
|
|
:toctree: dataset
|
2022-02-11 17:34:24 +08:00
|
|
|
|
:nosignatures:
|
|
|
|
|
|
:template: classtemplate_inherited.rst
|
2021-11-29 11:36:30 +08:00
|
|
|
|
|
2022-02-11 15:23:20 +08:00
|
|
|
|
mindspore.dataset.CSVDataset
|
|
|
|
|
|
mindspore.dataset.MindDataset
|
2022-03-25 15:34:09 +08:00
|
|
|
|
mindspore.dataset.OBSMindDataset
|
2022-02-11 15:23:20 +08:00
|
|
|
|
mindspore.dataset.TFRecordDataset
|
2021-11-29 11:36:30 +08:00
|
|
|
|
|
2022-02-11 09:07:57 +08:00
|
|
|
|
用户自定义
|
|
|
|
|
|
----------
|
2021-11-29 11:36:30 +08:00
|
|
|
|
|
2022-02-15 16:22:46 +08:00
|
|
|
|
.. mscnautosummary::
|
2021-11-29 11:36:30 +08:00
|
|
|
|
:toctree: dataset
|
2022-02-11 17:34:24 +08:00
|
|
|
|
:nosignatures:
|
|
|
|
|
|
:template: classtemplate_inherited.rst
|
2021-11-29 11:36:30 +08:00
|
|
|
|
|
2021-12-04 11:21:09 +08:00
|
|
|
|
mindspore.dataset.GeneratorDataset
|
2021-11-29 11:36:30 +08:00
|
|
|
|
mindspore.dataset.NumpySlicesDataset
|
|
|
|
|
|
mindspore.dataset.PaddedDataset
|
2022-02-11 17:34:24 +08:00
|
|
|
|
mindspore.dataset.RandomDataset
|
2021-11-29 11:36:30 +08:00
|
|
|
|
|
2022-02-11 15:23:20 +08:00
|
|
|
|
图
|
|
|
|
|
|
---
|
|
|
|
|
|
|
2022-02-15 16:22:46 +08:00
|
|
|
|
.. mscnautosummary::
|
2022-02-11 15:23:20 +08:00
|
|
|
|
:toctree: dataset
|
2022-03-25 15:34:09 +08:00
|
|
|
|
|
2022-07-16 10:58:06 +08:00
|
|
|
|
mindspore.dataset.ArgoverseDataset
|
|
|
|
|
|
mindspore.dataset.Graph
|
2022-02-11 15:23:20 +08:00
|
|
|
|
mindspore.dataset.GraphData
|
2022-07-16 10:58:06 +08:00
|
|
|
|
mindspore.dataset.InMemoryGraphDataset
|
2022-02-11 15:23:20 +08:00
|
|
|
|
|
2022-02-11 09:07:57 +08:00
|
|
|
|
采样器
|
2021-11-29 11:36:30 +08:00
|
|
|
|
-------
|
|
|
|
|
|
|
2022-02-15 16:22:46 +08:00
|
|
|
|
.. mscnautosummary::
|
2021-11-29 11:36:30 +08:00
|
|
|
|
:toctree: dataset
|
|
|
|
|
|
|
|
|
|
|
|
mindspore.dataset.DistributedSampler
|
|
|
|
|
|
mindspore.dataset.PKSampler
|
|
|
|
|
|
mindspore.dataset.RandomSampler
|
|
|
|
|
|
mindspore.dataset.SequentialSampler
|
|
|
|
|
|
mindspore.dataset.SubsetRandomSampler
|
|
|
|
|
|
mindspore.dataset.SubsetSampler
|
|
|
|
|
|
mindspore.dataset.WeightedRandomSampler
|
|
|
|
|
|
|
2021-12-04 11:21:09 +08:00
|
|
|
|
|
2022-02-11 09:07:57 +08:00
|
|
|
|
其他
|
|
|
|
|
|
-----
|
2021-11-29 11:36:30 +08:00
|
|
|
|
|
2022-02-15 16:22:46 +08:00
|
|
|
|
.. mscnautosummary::
|
2021-11-29 11:36:30 +08:00
|
|
|
|
:toctree: dataset
|
2022-02-11 17:34:24 +08:00
|
|
|
|
:nosignatures:
|
|
|
|
|
|
:template: classtemplate_inherited.rst
|
2021-11-29 11:36:30 +08:00
|
|
|
|
|
2022-02-11 17:34:24 +08:00
|
|
|
|
mindspore.dataset.BatchInfo
|
2021-12-04 11:21:09 +08:00
|
|
|
|
mindspore.dataset.DatasetCache
|
2022-02-11 15:23:20 +08:00
|
|
|
|
mindspore.dataset.DSCallback
|
2022-02-11 17:34:24 +08:00
|
|
|
|
mindspore.dataset.SamplingStrategy
|
2021-11-29 11:36:30 +08:00
|
|
|
|
mindspore.dataset.Schema
|
2022-02-11 17:34:24 +08:00
|
|
|
|
mindspore.dataset.Shuffle
|
2021-12-04 11:21:09 +08:00
|
|
|
|
mindspore.dataset.WaitedDSCallback
|
2022-02-11 17:34:24 +08:00
|
|
|
|
mindspore.dataset.OutputFormat
|
2021-12-04 11:21:09 +08:00
|
|
|
|
mindspore.dataset.compare
|
|
|
|
|
|
mindspore.dataset.deserialize
|
2021-11-29 11:36:30 +08:00
|
|
|
|
mindspore.dataset.serialize
|
|
|
|
|
|
mindspore.dataset.show
|
2022-03-29 13:19:46 +08:00
|
|
|
|
mindspore.dataset.sync_wait_for_dataset
|
2021-11-29 11:36:30 +08:00
|
|
|
|
mindspore.dataset.utils.imshow_det_bbox
|
|
|
|
|
|
mindspore.dataset.zip
|