You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

mindspore.dataset.Cifar10Dataset.rst 5.1 kB

4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100
  1. mindspore.dataset.Cifar10Dataset
  2. ================================
  3. .. py:class:: mindspore.dataset.Cifar10Dataset(dataset_dir, usage=None, num_samples=None, num_parallel_workers=None, shuffle=None, sampler=None, num_shards=None, shard_id=None, cache=None)
  4. 读取和解析CIFAR-10数据集的源文件构建数据集。该API目前仅支持解析二进制版本的CIFAR-10文件(CIFAR-10 binary version)。
  5. 生成的数据集有两列: `[image, label]` 。 `image` 列的数据类型是uint8。`label` 列的数据类型是uint32。
  6. **参数:**
  7. - **dataset_dir** (str): 包含数据集文件的根目录路径。
  8. - **usage** (str, 可选): 指定数据集的子集,可取值为'train','test'或'all'。
  9. 取值为'train'时将会读取50,000个训练样本,取值为'test'时将会读取10,000个测试样本,取值为'all'时将会读取全部60,000个样本。默认值:None,全部样本图片。
  10. - **num_samples** (int, 可选) - 指定从数据集中读取的样本数,可以小于数据集总数。默认值:None,读取全部样本图片。
  11. - **num_parallel_workers** (int, 可选) - 指定读取数据的工作线程数。默认值:None,使用mindspore.dataset.config中配置的线程数。
  12. - **shuffle** (bool, 可选) - 是否混洗数据集。默认值:None,下表中会展示不同参数配置的预期行为。
  13. - **sampler** (Sampler, 可选) - 指定从数据集中选取样本的采样器,默认值:None,下表中会展示不同配置的预期行为。
  14. - **num_shards** (int, 可选) - 指定分布式训练时将数据集进行划分的分片数,默认值:None。指定此参数后, `num_samples` 表示每个分片的最大样本数。
  15. - **shard_id** (int, 可选) - 指定分布式训练时使用的分片ID号,默认值:None。只有当指定了 `num_shards` 时才能指定此参数。
  16. - **cache** (DatasetCache, 可选) - 单节点数据缓存服务,用于加快数据集处理,详情请阅读 `单节点数据缓存 <https://www.mindspore.cn/docs/programming_guide/zh-CN/master/cache.html>`_ 。默认值:None,不使用缓存。
  17. **异常:**
  18. - **RuntimeError:** `dataset_dir` 路径下不包含数据文件。
  19. - **RuntimeError:** `num_parallel_workers` 参数超过系统最大线程数。
  20. - **RuntimeError:** 同时指定了 `sampler` 和 `shuffle` 参数。
  21. - **RuntimeError:** 同时指定了 `sampler` 和 `num_shards` 参数。
  22. - **RuntimeError:** 指定了 `num_shards` 参数,但是未指定 `shard_id` 参数。
  23. - **RuntimeError:** 指定了 `shard_id` 参数,但是未指定 `num_shards` 参数。
  24. - **ValueError:** `shard_id` 参数错误(小于0或者大于等于 `num_shards` )。
  25. .. note:: 此数据集可以指定参数 `sampler` ,但参数 `sampler` 和参数 `shuffle` 的行为是互斥的。下表展示了几种合法的输入参数组合及预期的行为。
  26. .. list-table:: 配置 `sampler` 和 `shuffle` 的不同组合得到的预期排序结果
  27. :widths: 25 25 50
  28. :header-rows: 1
  29. * - 参数 `sampler`
  30. - 参数 `shuffle`
  31. - 预期数据顺序
  32. * - None
  33. - None
  34. - 随机排列
  35. * - None
  36. - True
  37. - 随机排列
  38. * - None
  39. - False
  40. - 顺序排列
  41. * - `sampler` 实例
  42. - None
  43. - 由 `sampler` 行为定义的顺序
  44. * - `sampler` 实例
  45. - True
  46. - 不允许
  47. * - `sampler` 实例
  48. - False
  49. - 不允许
  50. **关于CIFAR-10数据集:**
  51. CIFAR-10数据集由60000张32x32彩色图片组成,总共有10个类别,每类6000张图片。有50000个训练样本和10000个测试样本。10个类别包含飞机、汽车、鸟类、猫、鹿、狗、青蛙、马、船和卡车。
  52. 以下为原始CIFAR-10数据集的结构,您可以将数据集文件解压得到如下的文件结构,并通过MindSpore的API进行读取。
  53. .. code-block::
  54. .
  55. └── cifar-10-batches-bin
  56. ├── data_batch_1.bin
  57. ├── data_batch_2.bin
  58. ├── data_batch_3.bin
  59. ├── data_batch_4.bin
  60. ├── data_batch_5.bin
  61. ├── test_batch.bin
  62. ├── readme.html
  63. └── batches.meta.text
  64. **引用:**
  65. .. code-block::
  66. @techreport{Krizhevsky09,
  67. author = {Alex Krizhevsky},
  68. title = {Learning multiple layers of features from tiny images},
  69. institution = {},
  70. year = {2009},
  71. howpublished = {http://www.cs.toronto.edu/~kriz/cifar.html}
  72. }
  73. .. include:: mindspore.dataset.Dataset.add_sampler.rst
  74. .. include:: mindspore.dataset.Dataset.rst
  75. .. include:: mindspore.dataset.Dataset.d.rst
  76. .. include:: mindspore.dataset.Dataset.use_sampler.rst
  77. .. include:: mindspore.dataset.Dataset.zip.rst