KerasのImageDataGeneratorを使用したデータ拡張時に、元画像のラベル情報をファイル名に埋め込みたい

Question

前提・実現したいこと

KerasのImageDataGeneratorを使用したデータ拡張時に、元画像のラベル情報をファイル名に埋め込みたい

画像認識系のタスクで、元画像データ(1,000枚)をデータ拡張して10,000枚増やしたいと考えています。
元画像データにはそれぞれラベル(仮にA,B,C,D...Z)がついています。

データ拡張の結果として出力される画像ファイル名の中に、ラベル情報を埋め込みたいのですが、うまくいかない状況です。。

例）
元画像ファイル名(data[0]： 123.png)
元画像に該当するラベル(labels[0]: 'A')　の場合、
データ拡張後の画像ファイル名: generated_123_A_01.png, generated_123_A_02.png, ...

発生している問題・エラーメッセージ

以下のようなコードで実行してみているのですが、★★★の部分を工夫することで、上記のことは実現できるものでしょうか..？
もしアイデア・方法をお持ちの方がいらっしゃいましたら、ご教示頂けますと大変助かります。

該当のソースコード

data = data[:1000]
labels = labels[:1000]
(中略)
datagen = ImageDataGenerator(
        rescale=1./255,
        zoom_range=0.2)
(中略)
g = datagen.flow(data, labels, batch_size=10, save_to_dir="../1_data_augmentation", save_format='png', save_prefix='generated_'+str(★★★))
for i in range(num_image):
    batches = g.next()

Answer 1

ソースコードを見ると、ImageDataGenerator のファイル名は以下のように決まるので、質問内容の事を ImageDataGenerator 単体でやるのは無理です。

<save_prefix で指定した値>_<画像のインデックス>_<7桁の乱数>.<拡張子>

keras-preprocessing/iterator.py at master · keras-team/keras-preprocessing

代替案として、ディレクトリごとにジェネレーターを作成して保存する方法があります。
この方法で名前にクラスIDを付与できますが、上記のように名前のフォーマットは決まっているので、 generated_123_A_01.png という形にはできないです。
どうしてもその名前にしたい場合はあとからリネームするか、Keras のソースコードを編集するのが手っ取り早いでしょう。

import numpy as np

# 保存するディレクトリ
save_dir = Path("generated")
save_dir.mkdir(exist_ok=True)

# -5° ~ 5° の範囲でランダムにせん断する。
datagen = image.ImageDataGenerator(shear_range=5)

for label in np.unique(labels):
    # ラベルが label の画像一覧
    label_imgs = imgs[labels == label]

    # ファイル名は label<クラスID>_<画像のインデックス>_<7桁の乱数>.<拡張子> となる。
    gen = datagen.flow(
        label_imgs,
        batch_size=1,
        save_to_dir=save_dir,
        save_prefix=f"label{label}",
        save_format="png",
    )
    for i in range(len(label_imgs)):
        gen.next()  # 一枚生成する。