我有一个Pandas DataFrame，其中包含大约10k张100x100的灰度图像（作为一个（100，100，1）numpy数组），沿着每个图像的标签（作为一个字符串，代表某人的名字）。我想将其转换为Keras数据集。
我读到我可以简单地使用dataset = tensorflow.data.Dataset.from_tensor_slices(dict(dataframe))，但它不起作用，可能是因为它是一个3D numpy数组？我是否应该将每个100x100的图像重新整形为一个10，000长的数组？
我正在尝试构建类似于CIFAR10的数据集。
1条答案
按热度按时间mxg2im7a1#
如果你能把标签从图像中分离出来，然后把图像 Dataframe 和标签 Dataframe 分别传递到一个元组中的
from_tensor_slices函数，它应该能工作...我试图通过生成形状为（10000，100，100，1）的随机值作为图像数据集来重新创建您的问题，在我看来效果不错。