首页 > 解决方案 > Keras Sequence ImageDatagenerator 在一个数组和两个数组中返回 X,y 时的性能差异

问题描述

我观察到一个奇怪的性能差异,并很想知道它为什么会发生。

我写了一个非常简单的自定义数据生成器,在 keras 中扩展了Sequence 类。

现在,当我在两个单独的数组中返回 X (图像)和 y (整数标签)时,我的纪元时间大约是返回时的 2 倍np.stack([X, y])。从源头上看,似乎没有关系。

这是由于 fit 函数调用生成器的某种方式造成的吗?

感谢您的帮助!

标签: kerastensorflow-datasets

解决方案


推荐阅读