Load ImageNet Base Model

We encourage you to start with a model pretrained on ImageNet. Load the model with the correct weights, set an input shape, and choose to remove the last layers of the model. Remember that images have three dimensions: a height, and width, and a number of channels. Because these pictures are in color, there will be three channels for red, green, and blue. We've filled in the input shape for you. This cannot be changed or the assessment will fail. If you need a reference for setting up the pretrained model, please take a look at notebook $05$b where we implemented transfer learning.

from tensorflow import keras

$$

base$\_$model $=$ keras.applications.VGG$16($

weights$=$'imagenet',

input$\_$shape$=(224,224,3),$

include$\_$top$=$False$)$