Image Modeling with Keras
Ariel Rokem
Senior Data Scientist, University of Washington
model = Sequential()
model.add(Conv2D(10, kernel_size=2, activation='relu',
input_shape=(img_rows, img_cols, 1)))
model.add(Flatten())
model.add(Dense(3, activation='softmax'))
model = Sequential() model.add(Conv2D(10, kernel_size=2, activation='relu', input_shape=(img_rows, img_cols, 1), padding='equal'))
# Second convolutional layer model.add(Conv2D(10, kernel_size=2, activation='relu')
model.add(Flatten()) model.add(Dense(3, activation='softmax'))
Image Modeling with Keras