Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- from keras.layers import Conv2D, MaxPooling2D, GlobalAveragePooling2D
- from keras.layers import Dropout, Flatten, Dense
- from keras.layers.advanced_activations import LeakyReLU
- from keras.models import Sequential
- model = Sequential()
- ### TODO: Define your architecture.
- model.add(Conv2D(16, (3,3), input_shape=(224, 224, 3), activation = "relu"))
- model.add(MaxPooling2D(pool_size=(2,2)))
- model.add(Conv2D(32, (3,3), input_shape=(224, 224, 3), activation = "relu"))
- model.add(MaxPooling2D(pool_size=(2,2)))
- model.add(Conv2D(64, (3,3), input_shape=(224, 224, 3), activation = "relu"))
- model.add(MaxPooling2D(pool_size=(2,2)))
- model.add(GlobalAveragePooling2D())
- model.add(Dense(133, activation='softmax'))
- model.summary()
Add Comment
Please, Sign In to add comment