Skip to content

Instantly share code, notes, and snippets.

@securetorobert

securetorobert/base_model.py

Created September 28, 2019 17:14
Show Gist options
  • Save securetorobert/e3635edeae6d8ec3437aa53294c35aa0 to your computer and use it in GitHub Desktop.
Save securetorobert/e3635edeae6d8ec3437aa53294c35aa0 to your computer and use it in GitHub Desktop.
Download ZIP
Create a base model for transfer learning in TF 2.0
Raw
base_model.py
base_model = tf.keras.applications.vgg19.VGG19(input_shape=(HEIGHT, WIDTH, NUM_CHANNELS), include_top=False, weights='imagenet')
base_model.trainable = False
print(base_model.summary())
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment