如何重塑来自datagen.flow_from_directory的输入？

我正在尝试使用需要以时间[时间步长，宽度，长度，通道]形式输入的时间分布的cnn层，但是flow_from_directory将仅提供每个图像作为输入，即[宽度，长度，通道]。我想重塑形状并作为输入。

我尝试以所需的格式创建所有数据的.npy，但是它在拟合过程中占用了所有内存，因此需要成块地传输数据，而不是一次传输所有内容。

是否可以选择部分使用数据？

我计划使用时间分布的cnn层和lstms对视频进行分类。（我没有写完整的东西，因为这不是问题，我希望您了解我想做的事情）

任何帮助将不胜感激。预先感谢。