Keras flow_from_directory 限制示例数量

Question

我可以flow_from_directory在 Keras 中使用的最简单方法是什么，同时将每个子目录中使用的示例数量限制为某个数字N？

对于上下文，我希望能够使用总图像的一小部分进行测试，而不必为较小的数据集创建单独的顶级目录，因为我在训练期间从 AWS S3 存储桶中提取这些数据。

Answer 1

新答案 编辑

正如评论中所指出的，应该使用已弃用的tf.keras.preprocessing.image_dataset_from_directory纯 Keras 等效项ImageDataGenerator。

旧答案

keras.preprocessing.image.ImageDataGenerator使用指定为 float 的参数创建validation_split。在这种情况下，您可以使用参数subsetinflow_from_directory从每个目录中仅获取一些样本。更多信息请点击这里。

如果您N特别想要每个文件夹中的图像，则必须计算每个目录中有多少个文件，并相应地设置训练验证分割。