Mastering Computer Vision with TensorFlow 2.x

Input image

In this section, we define the train and validation directory paths and use the os.path.join feature and the class name to define the directories within the train and validation directories. After that, we calculate the total number of images in each class directory using the len command:

train_dir = 'furniture_images/train'train_bed_dir = os.path.join(train_dir, 'bed')num_bed_train = len(os.listdir(train_bed_dir))

The total number of images in the train directories is obtained by summing the number of images in each class. The same method is applied for validation, which results in the output at the end being the total number of images in the train and validation directory. This information will be used by the neural network ...