In that case, you may want to write a function, which wraps the output of flow_from_dataframe and yield the reshaped array which should be used as the generator.