Skip to content

Instantly share code, notes, and snippets.

@jkmackie

jkmackie/load_VGG16.py

Created October 1, 2022 04:28
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save jkmackie/938eed48b9c51c0b534b850261708eb9 to your computer and use it in GitHub Desktop.
Save jkmackie/938eed48b9c51c0b534b850261708eb9 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
load_VGG16.py
# Load and freeze VGG16 model.
# Include_top=False removes classification layer trained on ImageNet.
base_model = VGG16(weights="imagenet", include_top=False, input_shape=(224,224,3))
base_model.trainable = False ## Not trainable weights
base_model.summary()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment