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.2 kB

4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101
  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. - **ValueError** - `num_parallel_workers` 参数超过系统最大线程数。
  20. - **ValueError** - `usage` 参数取值不为'train'、'test'或'all'。
  21. - **RuntimeError** - 同时指定了 `sampler` 和 `shuffle` 参数。
  22. - **RuntimeError** - 同时指定了 `sampler` 和 `num_shards` 参数或同时指定了 `sampler` 和 `shard_id` 参数。
  23. - **RuntimeError** - 指定了 `num_shards` 参数,但是未指定 `shard_id` 参数。
  24. - **RuntimeError** - 指定了 `shard_id` 参数,但是未指定 `num_shards` 参数。
  25. - **ValueError** - `shard_id` 参数错误(小于0或者大于等于 `num_shards` )。
  26. .. note:: 此数据集可以指定参数 `sampler` ,但参数 `sampler` 和参数 `shuffle` 的行为是互斥的。下表展示了几种合法的输入参数组合及预期的行为。
  27. .. list-table:: 配置 `sampler` 和 `shuffle` 的不同组合得到的预期排序结果
  28. :widths: 25 25 50
  29. :header-rows: 1
  30. * - 参数 `sampler`
  31. - 参数 `shuffle`
  32. - 预期数据顺序
  33. * - None
  34. - None
  35. - 随机排列
  36. * - None
  37. - True
  38. - 随机排列
  39. * - None
  40. - False
  41. - 顺序排列
  42. * - `sampler` 实例
  43. - None
  44. - 由 `sampler` 行为定义的顺序
  45. * - `sampler` 实例
  46. - True
  47. - 不允许
  48. * - `sampler` 实例
  49. - False
  50. - 不允许
  51. **关于CIFAR-10数据集:**
  52. CIFAR-10数据集由60000张32x32彩色图片组成,总共有10个类别,每类6000张图片。有50000个训练样本和10000个测试样本。10个类别包含飞机、汽车、鸟类、猫、鹿、狗、青蛙、马、船和卡车。
  53. 以下为原始CIFAR-10数据集的结构,您可以将数据集文件解压得到如下的文件结构,并通过MindSpore的API进行读取。
  54. .. code-block::
  55. .
  56. └── cifar-10-batches-bin
  57. ├── data_batch_1.bin
  58. ├── data_batch_2.bin
  59. ├── data_batch_3.bin
  60. ├── data_batch_4.bin
  61. ├── data_batch_5.bin
  62. ├── test_batch.bin
  63. ├── readme.html
  64. └── batches.meta.text
  65. **引用:**
  66. .. code-block::
  67. @techreport{Krizhevsky09,
  68. author = {Alex Krizhevsky},
  69. title = {Learning multiple layers of features from tiny images},
  70. institution = {},
  71. year = {2009},
  72. howpublished = {http://www.cs.toronto.edu/~kriz/cifar.html}
  73. }
  74. .. include:: mindspore.dataset.Dataset.add_sampler.rst
  75. .. include:: mindspore.dataset.Dataset.rst
  76. .. include:: mindspore.dataset.Dataset.d.rst
  77. .. include:: mindspore.dataset.Dataset.use_sampler.rst
  78. .. include:: mindspore.dataset.Dataset.zip.rst