使用文件名数据集,将图像数据集创建为元组

时间:2020-03-02 22:16:38

标签: python tensorflow

我在一个文件夹中创建了许多图像文件名的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,但均无效。

0 个答案:

没有答案