Skip to content

Instantly share code, notes, and snippets.

@NMZivkovic

NMZivkovic/base_models.py

Created November 10, 2019 20:20
Show Gist options
  • Save NMZivkovic/de895d1c4eae2493cd4b6d28369e8e23 to your computer and use it in GitHub Desktop.
Save NMZivkovic/de895d1c4eae2493cd4b6d28369e8e23 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
base_models.py
vgg16_base = tf.keras.applications.VGG16(input_shape=IMG_SHAPE, include_top=False, weights='imagenet')
googlenet_base = tf.keras.applications.InceptionV3(input_shape=IMG_SHAPE, include_top=False, weights='imagenet')
resnet_base = tf.keras.applications.ResNet101V2(input_shape=IMG_SHAPE, include_top=False, weights='imagenet')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment