自定义Keras数据生成器，产量高

自定义Keras数据生成器，产量高

您正在使用Sequence API，该API与普通生成器的工作原理有所不同。在生成器函数中，您将使用

yield

关键字在循环内执行迭代

whileTrue:

，因此，每次Keras调用生成器时，它都会获取一批数据，并自动环绕数据的末尾。

但是在序列中，函数有一个

index

参数

__getitem__

，因此不需要迭代或不需要迭代

yield

，这由Keras为您执行。这样可以使序列可以使用多重处理并行运行，而这对于旧的生成器函数是不可能的。

因此，您以正确的方式行事，无需任何更改。