如何从 tensorflow datasets 数据集中获取字符串映射？

Question

抱歉，如果这有一个明显的答案，我觉得应该。 使用 Tensorflow 数据集时，标签自动为整数，但您可以通过 API 显示它们以获取字符串映射。 例如在 CIFAR10 中 7 对应于马。 我怎样才能得到这些映射？ 我是否必须下载数据集并从文件夹/目录名称中获取它们？ 还是有别的办法？

干杯

编辑：正如 Amel 在下面指出的，我认为它必须以某种方式访问​​的原因是因为在调用显示示例时，类名显示在整数标签旁边。

Answer 1

你需要使用 train_ds.class_names（train_ds 是数据集）

例如：

class_names = train_ds.class_names
for images, labels in train_ds.take(1):
    print(labels)
    for i in range(0, 5):
        print(labels[i])
        print(class_names[labels[i]])

这里可以看到labels[i]是一个带整数的张量，class_names[labels[i]]是对应的字符串。