!2107 Fix CocoDataset docstring

Merge pull request !2107 from xiefangqi/xfq-coco-docstring
pull/2107/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
commit aae2e410ba

@ -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], - task='Detection', column: [['image', dtype=uint8], ['bbox', dtype=float32], ['category_id', dtype=uint32],
['iscrowd', dtype=uint32]]. ['iscrowd', dtype=uint32]].
task = 'Stuff' : column [['image', dtype=uint8], ['segmentation',dtype=float32], ['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='Keypoint', column: [['image', dtype=uint8], ['keypoints', dtype=float32],
task = 'Panoptic' : column [['image', dtype=uint8], ['bbox', dtype=float32], ['category_id', dtype=uint32], ['num_keypoints', dtype=uint32]].
['iscrowd', dtype=uint32], ['area', 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