我在一个文件夹中创建了许多图像文件名的tensorflow数据集。这些图像命名为[index] .jpg,其中index是用于标识图像的某个整数。我有一个字符串'index'的字典,以标签为元组。如何使用tf.data.Dataset.map将索引映射到标签元组?
这是我要传递给map函数的map_func:
def grabImages(filepath):
index = getIndexFromFilePath(filepath)
img = tf.io.read_file(filepath)
img = translateImage(img)
dictionary = getLabelDictionary()
return index, img
其中字典是标签dict的索引,索引是文件路径的索引,其格式为tf。Tensor和img是位于文件路径的预处理图像。
这将返回一个索引为张量的数据集,映射到相应的图像。是否可以使用index
之类的方法使用dictionary
获取dictionary[index]
的标签?基本上,我想找到index的字符串内容。
我尝试在.numpy()
函数内的当前会话中使用.eval()
和grabImages
,但均无效。