Skip to content

Instantly share code, notes, and snippets.

@NMZivkovic

NMZivkovic/models.py

Created November 10, 2019 20:28
Show Gist options
  • Save NMZivkovic/7d9554f162c1506b9a882151714c63e3 to your computer and use it in GitHub Desktop.
Save NMZivkovic/7d9554f162c1506b9a882151714c63e3 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
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