Skip to content

Instantly share code, notes, and snippets.

@weklund

weklund/model_architecture.py

Last active November 11, 2018 16:39
Show Gist options
  • Save weklund/12e20ecb2d00d38d240bc34c6b4a15d7 to your computer and use it in GitHub Desktop.
Save weklund/12e20ecb2d00d38d240bc34c6b4a15d7 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
model_architecture.py
model = InceptionResNetV2(weights = 'imagenet', include_top = False)
model = Sequential()
model.add(GlobalAveragePooling2D(input_shape = train_features_inception.shape[1:]))
model.add(Dropout(0.2))
model.add(Dense(1024, activation = 'relu'))
model.add(Dropout(0.2))
model.add(Dense(512, activation = 'relu'))
model.add(Dropout(0.2))
model.add(Dense(128, activation = 'relu'))
model.add(Dropout(0.2))
model.add(Dense(3, activation ='softmax'))
model.summary()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment