de_dataset = de.ImageFolderDataset(data_dir, num_parallel_workers=num_parallel_workers, shuffle=shuffle, sampler=sampler, class_indexing=class_indexing, num_shards=group_size, shard_id=rank)复制
上面是导入训练集的代码,class_indexing为None
class_indexing默认为None时,
class_indexing (dict): A str-to-int mapping from folder name to index (default=None, the folder names will be sorted alphabetically and each class will be given a unique index starting from 0).
想要获得类似{0: 'cat' , 1: 'dog'} 的class_index,
查阅官网后使用 class_indexing = de_dataset.get_col_names() 获取class index 无法得到结果,请问有什么api可以调用得到?
可以使用get_class_index这个api:
https://www.mindspore.cn/docs/api/zh-CN/r1.5/api_python/dataset/mindspore.dataset.ImageFolderDataset.html#mindspore.dataset.ImageFolderDataset.get_class_indexing