tensorflow - 如何使用 Tensorflow 平衡图像数据集

Question

我有一个不平衡的图像数据集。目前我正在使用ImageDataGenerator对象从目录中获取数据并适合 tensorflow 模型，fit_generator()但我想在将其输入模型之前平衡该类。

所以为此我创建了这个data_augmentation层来生成更多图像并将图像保存在文件夹中。然后用于ImageDataGenerator从文件夹中获取图像。

data_augmentation = Sequential([
                                layers.RandomFlip(
                                                  mode = "horizontal_and_vertical"
                                                  ),
                                layers.RandomRotation(0.5),
                                layers.RandomZoom(
                                                  height_factor = (0.2, 0.3),
                                                  width_factor= (0.2, 0.3) 
                                                  ),
                                layers.RandomTranslation( 
                                                        height_factor = 0.2, 
                                                        width_factor = 0.2
                                                         )
                                ])

但是要花很多时间来平衡班级。有没有另一种方法可以更快地做到这一点？