提交 a70da6ad 编写于 作者: X xiefangqi

fix CocoDataset docstring

上级 8de8289c
...@@ -4045,15 +4045,17 @@ class CocoDataset(MappableDataset): ...@@ -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], - 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.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册