Skip to content

Instantly share code, notes, and snippets.

@Higgcz
Created September 21, 2016 09:02
Show Gist options
  • Save Higgcz/3e64a5c7a33a32d1ce91dbe9795c5dd1 to your computer and use it in GitHub Desktop.
Save Higgcz/3e64a5c7a33a32d1ce91dbe9795c5dd1 to your computer and use it in GitHub Desktop.
Keras 1.1.0 - Model
# Create simple model
model = Sequential()
model.add(TimeDistributed(Convolution2D(32, 3, 3, border_mode='same'), input_shape = batch_input_shape[1:]))
model.add(Activation('relu'))
model.add(TimeDistributed(MaxPooling2D(pool_size = (2,2))))
model.add(BatchNormalization())
model.add(TimeDistributed(Convolution2D(32, 3, 3, border_mode='same')))
model.add(Activation('relu'))
model.add(TimeDistributed(MaxPooling2D(pool_size = (2,2))))
model.add(BatchNormalization())
model.add(TimeDistributed(Convolution2D(32, 3, 3, border_mode='same')))
model.add(Activation('relu'))
model.add(TimeDistributed(MaxPooling2D(pool_size = (2,2))))
model.add(BatchNormalization())
model.add(TimeDistributed(Convolution2D(32, 2, 2, border_mode='same')))
model.add(Activation('relu'))
model.add(TimeDistributed(MaxPooling2D(pool_size = (2,2))))
model.add(BatchNormalization())
model.add(TimeDistributed(Flatten()))
model.add(LSTM(32, stateful = True, return_sequences = False))
model.add(Dense(1))
model.add(Activation('sigmoid'))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment