classification - 如何加载图像数据集并计算每个类中的图像数量？

Question

我正在使用 CNN 多类图像分类器项目。我的训练数据集包括 2 类图像。我想找出每个班级的图像数量。我附上了下面的代码。

#for training
train_dataset = train.flow_from_directory('/content/drive/MyDrive/System/training',
                                          target_size=(200,200), batch_size=32,
                                          class_mode='binary')