代码之家 › 专栏 › 技术社区 › Austin

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

generator keras

Austin · 技术社区 · 7 年前

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

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

1 回复 | 直到 7 年前

Szymon Maszke 7 年前

创造 keras.preprocessing.image.ImageDataGenerator 争论不休 validation_split 指定为浮动。在这种情况下,您可以使用参数 subset 在里面 flow_from_directory 仅从每个目录中获取一些示例。更多信息 here

如果你愿意 N 具体来说,您必须计算每个目录中有多少文件,并相应地设置train validation split。

推荐文章

albert · AttributeError:“str”对象没有属性“str”如何解决此问题

4 年前

CoccoSyn · Keras TypeError:其中一个输入没有可接受的类型。具有参差不齐张量的LSTM

4 年前

Joseph · 重塑BatchDataset训练模型的输入-Tensorflow

4 年前

Chris Ranck · Keras时间序列生成器-输入为“X”长,输出为“X-长”

4 年前

DPM · 将自定义掩码传递给LSTM数据以进行培训和验证

4 年前

Luciana Carota · 为什么Keras的R“预测”函数不能识别正确的输入维度?

4 年前

Sarah Elnaggar · 我使用Gensim Doc2vec进行图形嵌入,然后在keras中使用两层深度神经网络进行二元分类

4 年前

user836026 · 在pytorch中复制权重时语法无效

4 年前

mad · 如何从一个定制的神经网络模型中获得逻辑和概率

4 年前

Archit · 无法使用自定义距离函数加载带有lambda函数的keras模型

8 年前