Skip to content

Instantly share code, notes, and snippets.

@ajinkyajawale14499
Created July 8, 2019 19:40
Show Gist options
  • Save ajinkyajawale14499/8a60897fdf8b1a76c0ff8b73e0e2e25c to your computer and use it in GitHub Desktop.
Save ajinkyajawale14499/8a60897fdf8b1a76c0ff8b73e0e2e25c to your computer and use it in GitHub Desktop.
IMG_SHAPE = (IMAGE_SIZE, IMAGE_SIZE, 3)
# Create the base model from the pre-trained model MobileNet V2
base_model = tf.keras.applications.MobileNetV2(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