【发布时间】:2019-06-06 17:05:09
【问题描述】:
我可以在 Keras 中使用 flow_from_directory 的最简单方法是什么,同时将每个子目录中使用的示例数量限制为某个数量 N?
就上下文而言,我希望能够使用总图像的一小部分进行测试,而不必为较小的数据集创建单独的顶级目录,因为我从 AWS S3 存储桶中提取这些数据训练期间。
【问题讨论】:
我可以在 Keras 中使用 flow_from_directory 的最简单方法是什么,同时将每个子目录中使用的示例数量限制为某个数量 N?
就上下文而言,我希望能够使用总图像的一小部分进行测试,而不必为较小的数据集创建单独的顶级目录,因为我从 AWS S3 存储桶中提取这些数据训练期间。
【问题讨论】:
创建keras.preprocessing.image.ImageDataGenerator,参数validation_split 指定为浮点数。在这种情况下,您可以在flow_from_directory 中使用参数subset 来仅从每个目录中获取一些样本。更多信息here。
如果您想要每个文件夹中的N 图像,则必须计算每个目录中有多少文件,并相应地设置训练验证拆分。
【讨论】: