python - 如何知道flow_from_directory中的图像数量

Question

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

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