Guest User

Untitled

a guest
Nov 2nd, 2019
151
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 1.29 KB | None | 0 0
  1. batch_shape = (32, 64, 64, 3)
  2.  
  3. ipt = Input(batch_shape=batch_shape)
  4. x   = Conv2D(6,  (8, 8), strides=(1, 1), activation='relu',  padding='valid',
  5.                                kernel_initializer='he_normal', name='cnn_0')(ipt)
  6. x   = BatchNormalization(name='bn_0')(x)
  7. x   = Dropout(0.1)(x)
  8. x   = Conv2D(12, (8, 8), strides=(2, 2), activation='relu', padding='valid',
  9.                                kernel_initializer='he_normal', name='cnn_1')(x)
  10. x   = BatchNormalization(name='bn_1')(x)
  11. x   = Conv2D(24, (4, 4), strides=(2, 2), activation='relu', padding='valid',
  12.                                kernel_initializer='he_normal', name='cnn_2')(x)
  13. x   = BatchNormalization(name='bn_2')(x)
  14. x   = Conv2D(48, (3, 3), strides=(2, 2), activation='relu', padding='valid',
  15.                                kernel_initializer='he_normal', name='cnn_3')(x)
  16. x   = BatchNormalization(name='bn_3')(x)
  17. x   = Conv2D(12, (1, 1), strides=(1, 1), activation='relu', padding='valid',
  18.                                kernel_initializer='he_normal', name='cnn_4')(x)
  19. x   = BatchNormalization(name='bn_4')(x)
  20. x   = Flatten()(x)
  21. x   = Dense(12, activation='relu')(x)
  22. out = Dense(6, activation='softmax', name='output')(x)
  23.  
  24. model = Model(ipt, out)
  25. model.compile('adam', 'sparse_categorical_crossentropy', metrics=['accuracy'])
Add Comment
Please, Sign In to add comment