在keras中使用imagedatagenerator()后，训练前获取图片数量

from keras.preprocessing.image import ImageDataGenerator
import os

datagen = ImageDataGenerator()

train_data_dir = 'path_to_train_data_directory'

train_generator = datagen.flow_from_directory(
    train_data_dir,
    target_size=(image_width, image_height),
    batch_size=batch_size,
    class_mode='categorical',
    shuffle=False
)

num_images = len(train_generator.filenames)

from keras.preprocessing.image import ImageDataGenerator
import os

datagen = ImageDataGenerator()

train_data_dir = 'path_to_train_data_directory'

train_generator = datagen.flow_from_directory(
    train_data_dir,
    target_size=(image_width, image_height),
    batch_size=batch_size,
    class_mode='categorical',
    shuffle=False
)

num_images = len(train_generator.filenames)