Project Creation: Part Two

In this lesson, we will be performing the feature extractor strategy to build our classifier.

We made, shuffled, and preprocessed our dataset in the previous lesson. In this lesson, we will work with the ResNet50 model.

Importing the libraries

We will use TensorFlow's Keras module to build this project.

Press + to interact
from tensorflow.python.keras.applications.resnet50 import ResNet50
from tensorflow.python.keras.optimizers import Adam
from tensorflow.python.keras.layers import *
from tensorflow.python.keras.models import Model
import numpy as np
print("Imported Successfully!")

Load the ResNet50 model

The next step is to load the ResNet50 model. We will also have a look at the summary of the model to understand the architecture of the ResNet50 model.

Press + to interact
model = ResNet50(include_top = False, weights = 'imagenet', input_shape = (224,224,3))
print(model.summary())

Explanation:

  • The weights will start to download. It will take a bit of time to complete the download process. ...

Access this course and 1400+ top-rated courses and projects.