Keras - flow_from_directory:如何获取每个类别的样本数量？

from keras.preprocessing.image import ImageDataGenerator

# 定义图像生成器
datagen = ImageDataGenerator()

# 从目录中读取图像数据
data_generator = datagen.flow_from_directory(
    directory='path/to/dataset',
    target_size=(224, 224),
    batch_size=32,
    class_mode='categorical')

# 获取类别名称和对应的索引值
class_indices = data_generator.class_indices

# 遍历字典，获取每个类别的样本数量
for class_name, class_index in class_indices.items():
    # 使用class_index作为过滤条件，获取对应类别的样本数量
    num_samples = len(data_generator.labels[data_generator.labels == class_index])
    print(f"类别 {class_name} 的样本数量为 {num_samples} 个")