Browse Source

!2107 Fix CocoDataset docstring

Merge pull request !2107 from xiefangqi/xfq-coco-docstring
tags/v0.5.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
aae2e410ba
1 changed files with 9 additions and 7 deletions
  1. +9
    -7
      mindspore/dataset/engine/datasets.py

+ 9
- 7
mindspore/dataset/engine/datasets.py View File

@@ -4074,15 +4074,17 @@ class CocoDataset(MappableDataset):
A source dataset for reading and parsing COCO dataset. A source dataset for reading and parsing COCO dataset.


CocoDataset support four kinds of task: CocoDataset support four kinds of task:
2017 Train/Val/Test Detection, Keypoints, Stuff, Panoptic.
2017 Train/Val/Test Detection, Keypoints, Stuff, Panoptic.


The generated dataset has multi-columns : The generated dataset has multi-columns :
task = 'Detection' : column [['image', dtype=uint8], ['bbox', dtype=float32], ['category_id', dtype=uint32],
['iscrowd', dtype=uint32]].
task = 'Stuff' : column [['image', dtype=uint8], ['segmentation',dtype=float32], ['iscrowd',dtype=uint32]].
task = 'Keypoint' : column [['image', dtype=uint8], ['keypoints', dtype=float32], ['num_keypoints', dtype=uint32]].
task = 'Panoptic' : column [['image', dtype=uint8], ['bbox', dtype=float32], ['category_id', dtype=uint32],
['iscrowd', dtype=uint32], ['area', dtype=uint32]].
- task='Detection', column: [['image', dtype=uint8], ['bbox', dtype=float32], ['category_id', dtype=uint32],
['iscrowd', dtype=uint32]].
- task='Stuff', column: [['image', dtype=uint8], ['segmentation',dtype=float32], ['iscrowd',dtype=uint32]].
- task='Keypoint', column: [['image', dtype=uint8], ['keypoints', dtype=float32],
['num_keypoints', dtype=uint32]].
- task='Panoptic', column: [['image', dtype=uint8], ['bbox', dtype=float32], ['category_id', dtype=uint32],
['iscrowd', dtype=uint32], ['area', dtype=uint32]].

This dataset can take in a sampler. sampler and shuffle are mutually exclusive. Table This dataset can take in a sampler. sampler and shuffle are mutually exclusive. Table
below shows what input args are allowed and their expected behavior. below shows what input args are allowed and their expected behavior.




Loading…
Cancel
Save