Keras flow_from_directory限制示例数

Question

在Keras中使用flow_from_directory并将每个子目录中使用的示例数限制为一定数量N时，最简单的方法是什么？

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

Answer 1

使用参数keras.preprocessing.image.ImageDataGenerator指定为float来创建validation_split。在这种情况下，可以在subset中使用参数flow_from_directory从每个目录中仅获取一些样本。更多信息here。

如果要从每个文件夹中分别获取N张图像，则必须计算每个目录中有多少个文件，并相应地设置训练验证拆分。