gpt4 book ai didi

Keras flow_from_directory 限制示例数量

转载 作者:行者123 更新时间:2023-12-03 22:11:17 26 4
gpt4 key购买 nike

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

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

最佳答案

创建 keras.preprocessing.image.ImageDataGenerator带参数 validation_split指定为浮点数。在这种情况下,您可以使用参数 subsetflow_from_directory从每个目录中只获取一些样本。更多信息 here .

如果你想要 N具体来自每个文件夹的图像,您必须计算每个目录中有多少文件,并相应地设置训练验证拆分。

关于Keras flow_from_directory 限制示例数量,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/54152216/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com