如何获取TensorFlow Dataset(tfds)中的类别名称？

你用tfds.ImageFolder加载符合目录结构的数据集后，直接从数据集元信息中读取ClassLabel的names属性即可，操作如下：

1. 加载数据集（你已完成的步骤）

import tensorflow_datasets as tfds

# 替换为你的数据集根目录路径
builder = tfds.ImageFolder("./your_dataset_root")

2. 单命令获取类别名列表

class_names = builder.info.features["label"].names

说明 #

• 返回的class_names是按整数标签索引排序的列表，整数标签n对应的实际类别名称就是class_names[n]，和你数据集的二级子目录命名完全对应
• 可以直接将该列表传入绘图、分类报告、混淆矩阵等工具的对应参数使用，比如scikit-learn的classification_report的target_names参数可直接传入该变量

内容的提问来源于stack exchange，提问作者JKnecht