mindspore.dataset.Dataset.get_class_indexing

mindspore.dataset.Dataset.get_class_indexing()[源代码]

返回类别索引。

返回:

dict,描述类别名称到索引的键值对映射关系,通常为str-to-int格式。针对COCO数据集,类别名称到索引映射关系描述形式为str-to-list<int>格式,列表中的第二个数字表示超类别。

样例:

>>> # dataset is an instance object of Dataset
>>> class_indexing = dataset.get_class_indexing()