如何从tensorflow数据集中访问图像？

Question

我已经在张量流中创建了一个图像数据集，但现在我想使用数据集中的图像作为变量（以查看和操作它等），但我根本不知道如何访问它，因为批处理数据集不可下标，所以我写不出来

image=dataset[0]

我尝试在网上查找，但只找到了

take

命令，该命令似乎每次都会选择随机图像，这不是我想要的。任何帮助将不胜感激，提前致谢！

我尝试从张量流批处理数据集中访问图像，但我的命令不起作用，所以我正在寻求帮助。

Answer 1

好吧，我不是 TF/python 专家，但我最近遇到了类似的问题，这也可能对你有帮助。我创建了一个由输入和输出图像组成的 tf 数据集，我想查看它们以确保它们是正确的 - 为了做到这一点，我需要从 tf 数据集结构中提取图像，这基本上也是您所需要的我相信？虽然我不清楚您是否只需要一张图像或所有图像，但无论如何我相信您可以迭代 tf 数据集并提取数据部分，如下所示：

for img in my_dataset: 
    one_image = img.numpy()

如果您的数据集由图像和标签组成，那么

img

将是

(image, label)

的元组，因此您必须通过简单的索引来提取图像：

for img in my_dataset: 
    one_image = img[0].numpy()

希望有帮助。