Как узнать количество изображений в 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

Я хотел бы написать al oop для подсчета числа изображений на train_set

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

, но это ничего не отображает !!

Answer 1

Чтобы получить нет. изображений, попробуйте с приведенным ниже кодом.

train_set.samples 

Answer 2

__ iter __ будет вашим решением.

сначала вы попытаетесь предсказать тип train_set .Если вы не знаете.

print(type(train_set))
#then you find **keras.preprocessing.image.DirectoryIterator**

Теперь вы хотите применить некоторую функцию или модификацию к данным этого типа, перейдите по этой ссылке здесь