How knowing number of images in flow_from_directory

Question

We can generate image dataset using ImageDataGenerator with flow_from_directory method.

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(..)

and that displays:

Found 200 images belonging to 2 classes

I would like to write a loop for to count the number of images on train_set

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

but this does'nt display anything!!

Answer 1

To access the number of samples of your dataset you first must know which type it is:

If you are using the ImageDataGenerator then:

type(train_ds)

will return tensorflow.python.keras.preprocessing.image.DirectoryIterator. In this case you can access the number of samples with:

train_ds.samples

However, if you are creating your creating your dataset using:

train_ds = tf.keras.preprocessing.image_dataset_from_directory( rescale = 1/255. , etc...)

then

type(train_ds)

will return tensorflow.python.data.ops.dataset_ops.BatchDataset which means that you can access the number of samples indirectly with

len(train_ds.file_paths)

Answer 2

To get the no. of images, try with the below code.

train_set.samples