如何知道 flow_from_directory 中的图像数量

Question

我们可以使用 ImageDataGenerator 和 flow_from_directory 方法生成图像数据集。

train_datagen = ImageDataGenerator(
    rescale=1./255, #scale images from integers 0-255 to floats 0-1.
    shear_range=0.2,
    zoom_range=0.2, # zoom in or out in images
    horizontal_flip=True) #horizontal flip of images
train_set = train_datagen.flow_from_directory(..)

并显示：

Found 200 images belonging to 2 classes

我想编写一个循环来计算 train_set 上的图像数量

For image in train_set:
    count = count+1
print(count)

但这不显示任何东西！

Answer 1

得到没有。的图像，请尝试使用以下代码。

train_set.samples