ImageDataGeneraterを使用して増えた画像枚数を知りたい。

Question

前提・実現したいこと

ImageDataGeneraterを使用して増えた画像枚数を知りたい。
実際に画像をどのくらいに増やして実装したのかを示したいが、何枚に増えたのか分かりませんでした。

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

Found 1488 images belonging to 2 classes.

該当のソースコード

import tensorflow as tf
from tensorflow.keras.preprocessing.image import ImageDataGenerator

train_path = "path/train"

train_datagen_augmented = ImageDataGenerator(
     rescale=1/255,
     rotation_range=40,
     width_shift_range=0.2,
     height_shift_range=0.2,
     shear_range=0.2,
     zoom_range=0.2,
     horizontal_flip=True,
     fill_mode="nearest"
)

train_generator_augmented = train_datagen_augmented.flow_from_directory(
    train_path,
    target_size=(224, 224),
    batch_size=32,
    color_mode="rgb",
    class_mode="sparse",
    shuffle=True,
)

試したこと

いろいろ調べて見まして、生成画像を何枚か表示する方法などはありましたが、実際どのくらいに増えたのか分かりませんでした。
ぜひよろしくお願い致します。

補足情報（FW/ツールのバージョンなど）

GoogleColaboratoryを使用しています。

Answer 1

ImageDataGeneratorでデータを水増ししたいの回答を参照ください。
generatorオブジェクトを生成しただけでは水増しされたデータは作成されません。
それを走査することによりデータが生成されて返されます。
つまり走査しつづけるかぎり無限に生成することができますので事前に何個というのは分かりません。

Comments

[rt.]

ありがとうございます！
画像生成のされ方については理解できました。
その上で、学習後に生成された画像枚数の総数を取得する方法はないのでしょうか？
追加で申し訳ないです。

[can110]

学習処理およびそのパラメータによりますが
結局は単純に学習データ数（×エポック数）となるはずです。

[rt.]

ありがとうございます！