Using Keras ImageDataGenerator with Transfer Learning
This line of code is used to define the transformations that the training DataGenerator will apply on all the images to augment the size of the dataset. For the validation DataGenerator, we only specify the scaling factor. The other transformations are not required because we are not training the model on this data. Next, we define the Model. We set layer.trainable False for each layer of the VGG model, as we are using the pre-trained weights of the model.
Sep-7-2020, 08:05:10 GMT
- Technology: