Skip to content

Instantly share code, notes, and snippets.

@k26dr
Last active February 13, 2017 23:38
Show Gist options
  • Save k26dr/1470915ce70b143aca5b28c97b12e9aa to your computer and use it in GitHub Desktop.
Save k26dr/1470915ce70b143aca5b28c97b12e9aa to your computer and use it in GitHub Desktop.
Modified CNN based off AlexNet for Behavioral Cloning CarND project
model = Sequential()
# Output = 55 x 55
model.add(Lambda(lambda x: x / 255 - 0.5, input_shape=(227, 227, 3)))
model.add(Convolution2D(48, 11, 11, subsample=(4,4)))
model.add(Activation('relu'))
# Output = 27 x 27
model.add(MaxPooling2D((3,3), strides=(2,2)))
model.add(Convolution2D(128, 5, 5, border_mode='same'))
model.add(Activation('relu'))
# Output = 13 x 13
model.add(MaxPooling2D((3,3), strides=(2,2)))
model.add(Convolution2D(192, 3, 3, border_mode='same'))
model.add(Activation('relu'))
# Output = 13 x 13
model.add(Convolution2D(192, 3, 3, border_mode='same'))
model.add(Activation('relu'))
# Output = 13 x 13
model.add(Convolution2D(128, 3, 3, border_mode='same'))
model.add(Activation('relu'))
# Output = 4 x 4 x 128
model.add(MaxPooling2D((3,3)))
model.add(Flatten())
model.add(Dense(2048))
model.add(Dense(2048))
model.add(Dense(1))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment