From a70da6ad47a1252873d15aca4c7b73712bbd5655 Mon Sep 17 00:00:00 2001 From: xiefangqi Date: Mon, 15 Jun 2020 09:30:27 +0800 Subject: [PATCH] fix CocoDataset docstring --- mindspore/dataset/engine/datasets.py | 16 +++++++++------- 1 file changed, 9 insertions(+), 7 deletions(-) diff --git a/mindspore/dataset/engine/datasets.py b/mindspore/dataset/engine/datasets.py index 2beadbfff1..d5d93aa0ab 100644 --- a/mindspore/dataset/engine/datasets.py +++ b/mindspore/dataset/engine/datasets.py @@ -4045,15 +4045,17 @@ class CocoDataset(MappableDataset): A source dataset for reading and parsing COCO dataset. 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 : - 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 below shows what input args are allowed and their expected behavior.