Skip to content

Instantly share code, notes, and snippets.

@himanshurawlani
Last active March 21, 2019 12:36
Show Gist options
  • Save himanshurawlani/8864cb2b71cc568f8c05e44a161ac724 to your computer and use it in GitHub Desktop.
Save himanshurawlani/8864cb2b71cc568f8c05e44a161ac724 to your computer and use it in GitHub Desktop.
We can download Inception V3 base model pre-trained in Imagenet dataset using this script.
from tensorflow import keras
# Create the base model from the pre-trained model MobileNet V2
base_model = keras.applications.InceptionV3(input_shape=IMG_SHAPE,
# We cannot use the top classification layer of the pre-trained model as it contains 1000 classes.
# It also restricts our input dimensions to that which this model is trained on (default: 299x299)
include_top=False,
weights='imagenet')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment