gpt4 book ai didi

python - 类似于 ImageDataGenerator 迭代磁盘上的数组

转载 作者:行者123 更新时间:2023-11-30 09:42:25 27 4
gpt4 key购买 nike

我有 70'000 个 2D numpy 数组,我想在其上使用 Keras 训练 CNN 网络。将它们保存在内存中是一种选择,但会消耗大量内存。因此,我想将矩阵保存在磁盘上并在运行时加载它们。一种选择是使用 ImageDataGenerator。问题是它只能读取图像。

我想将数组存储为不作为图像,因为当我将它们保存为(灰度)图像时,数组的值会发生变化(标准化等)。但最后我想将原始矩阵输入到网络中,而不是由于保存为图像而改变值。

是否可以以某种方式将数组存储在磁盘上并以与 ImageDataGenerator 类似的方式迭代它们?

或者我可以将数组保存为图像而不更改数组的值吗?

最佳答案

您可以通过重写类的一些简单方法来定义自己的自定义数据生成器类,而不是使用 ImageDataGenerator。

您可以关注此媒体帖子以获取更多相关引用。

https://medium.com/@ensembledme/writing-custom-keras-generators-fe815d992c5a

关于python - 类似于 ImageDataGenerator 迭代磁盘上的数组,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/57047940/

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