Skip to content

Instantly share code, notes, and snippets.

@amankharwal
Created December 24, 2020 15:13
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
Star You must be signed in to star a gist
Save amankharwal/cafecb5e45fb3dcebdafd1ac5a325371 to your computer and use it in GitHub Desktop.
model = Sequential()
model.add(VGG16(weights="imagenet", include_top=False, input_shape=(HEIGHT, WIDTH, CHANNEL)))
model.add(Flatten())
model.add(Dense(128, activation="relu"))
model.add(Dense(64, activation="relu"))
model.add(Dense(64, activation="relu"))
model.add(Dense(4, activation="sigmoid"))
model.layers[-6].trainable = False
model.summary()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment