Keras flow_from_directory 返回什么？答案

【问题标题】：What does Keras flow_from_directory return?Keras flow_from_directory 返回什么？
【发布时间】：2021-10-11 00:55:37
【问题描述】：

image_gen = ImageDataGenerator(rescale=1.0/255, horizontal_flip=True)

train_data_gen = image_gen.flow_from_directory(batch_size=BATCH_SIZE, directory=train_dir, target_size=(IMG_SHAPE,IMG_SHAPE), shuffle=True)

augmented_images = [train_data_gen[0][0][0] for i in range(5)]

所以我最近开始研究 Tensorflow 并尝试创建自己的 CNN，同时查看如何使用 ImageDataGenerator 我遇到了这段代码。我想知道 .flow_from_directory 返回什么似乎是图像的可迭代对象。然而让我困惑的是为什么 train_data_gen 有三个维度，这些维度是什么意思。

【问题讨论】：

大概输出的3个维度代表(Batch_size,height,width)。

标签： tensorflow machine-learning image-processing keras conv-neural-network

【解决方案1】：

train_data_gen 就是这样，一个生成器。为了产生输出，需要执行代码

images, labels=next(train_data_gen)

结果是 batch_size 数量的图像及其相关标签。图片将具有形状（batch_size，IMG_SHAPE，IMG_SHAPE，通道），标签是形状（batch_size，1）。

【讨论】：