在Keras上使用ImageDataGenerator生成用于训练自动编码器的样本

Question

我正在开发一种自动编码器，以绘制正方形来界定具有特定特征的给定图像的区域。我的数据集很小（约有400个样本），我的工作之一是使用来自keras的ImageDataGenerator生成更多样本，但我遇到了麻烦。我需要为输入图像（即原始图像）生成更多数据（旋转，镜像等），并在输出中（即正方形位于正确位置的图像）生成这种数据，这样我就需要一种对两个图像应用相同变换以在输入图像和输出图像之间具有对应关系的方法。