Keras 有没有办法在使用 ImageDataGenerator 时获取每个类别中的图像数量？假设我有一个包含 3 个子文件夹（类）的 train 文件夹，每个子文件夹都包含一定数量的图像。
imagedatagen = ImageDataGenerator(...)
imageflow = imagedatagen.flow_from_directory(
source_dir,
class_mode='categorical', ...)
我可以使用以下命令获取类映射或所有样本的数量：
class_dict = imageflow.class_indices # returns ie. {'aligator': 0, 'cat': 1, 'dog': 2}
sample_count = imageflow.samples # returns sum of all samples, not divided into categories
但是我如何获得信息，类别 0 包含 1000 张图片，类别 1 包含 1500 张图片，类别 2 包含其余所有图片？我知道我可以在os模块的帮助下迭代所有文件，但想知道是否还有其他方法，但我找不到它。