Use a Pre-trained Model
In this lesson, we are going to use the first strategy of transfer learning i.e. using pre-trained models. We will use ResNet50 to build a classifier.
We'll cover the following
Here, we will use the ResNet50
Model. The pre-trained model can classify images into 1000 object categories, such as keyboard, mouse, pencil, animals, etc.
Note that 50 means that the network is 50 layers deep. You can also have a 101 or a 152 layers deep network.
Importing the required libraries
We will import the ResNet50
model from the Keras
library. There are many other pre-trained models in the keras.applications
module. Check them out here.
Get hands-on with 1400+ tech skills courses.