Как сохранить файл, добавив предыдущее имя в качестве префикса после ImageDataGenerator

Question

Использование ImageDataGenerator из Keras. предположим, что моя структура папок похожа на

      a -
  1.jpg
  2.jpg
  3.jpg
b -
  5.jpg
  6.jpg
  7.jpg

Я делаю увеличение, как показано ниже:

      for i in range (20):
    for label in LABELS:  # "LABELS" is the folder name here
        folder_tail = <--- -----------------------------------------how to ????
        if os.path.isdir('./f/'+folder_tail):
            print("Exists")
        else:
            print("Doesn't exists")
            os.makedirs('./f/'+folder_tail)

        datagen_kwargs = dict(rescale=1./255)  
        dataflow_kwargs = dict(target_size=(IMAGE_SIZE, IMAGE_SIZE), 
                            batch_size=BATCH_SIZE, interpolation="bilinear")

        train_datagen = tf.keras.preprocessing.image.ImageDataGenerator(
#         rotation_range=40,
#         horizontal_flip=True,
#         width_shift_range=0.2, 
#         height_shift_range=0.2,
#         shear_range=0.2, 
#         zoom_range=0.2,
#         fill_mode = 'nearest',
        **datagen_kwargs)

        train_generator = train_datagen.flow_from_directory(
          data_dir, subset="training", shuffle=True, save_to_dir='./f/'+ folder_tail, save_prefix=folder_tail , classes=[label], **dataflow_kwargs)

        batch = next(train_generator)

Он работает нормально, но я столкнулся с проблемой сохранения файла. Я хочу сохранить файлы с их прежним именем в качестве префиксов после процесса расширения.

нравиться

      a - 
1_0_786786.jpg
1_0_786786.jpg
1_0_786786.jpg
......
2_0_785686.jpg
2_0_786786.jpg
2_0_786786.jpg
....

Как я могу добавить свой folder_tail?