来自目录的Keras flow_限制了示例数量

来自目录的Keras flow_限制了示例数量,keras,generator,Keras,Generator,我在Keras中使用flow\u from\u directory的最简单方法是什么,同时将每个子目录中使用的示例数量限制在某个数量N 在上下文中,我希望能够使用总图像的一小部分进行测试,而不必为较小的数据集创建单独的顶级目录,因为我在训练期间从AWS S3存储桶中提取这些数据。创建keras.preprocessing.image.ImageDataGenerator,参数验证\u split指定为float。在这种情况下,您可以在flow\u from\u directory中使用参数sub

我在Keras中使用
flow\u from\u directory
的最简单方法是什么,同时将每个子目录中使用的示例数量限制在某个数量
N


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

创建
keras.preprocessing.image.ImageDataGenerator
,参数
验证\u split
指定为float。在这种情况下,您可以在
flow\u from\u directory
中使用参数
subset
,仅从每个目录中获取一些样本。更多信息

如果您想从每个文件夹中获取图像,则必须计算每个目录中有多少文件,并相应地设置train validation split