Browse Source

fix CocoDataset docstring

tags/v0.5.0-beta
xiefangqi 5 years ago
parent
commit
a70da6ad47
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

@@ -4045,15 +4045,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