Advertisement
Guest User

Untitled

a guest
Jun 26th, 2019
52
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.89 KB | None | 0 0
  1. def get_disc_normal(image_shape=(64,64,3)):
  2. dropout_prob = 0.4
  3. kernel_init = 'glorot_uniform'
  4. dis_input = Input(shape = image_shape)
  5.  
  6. # Conv layer 1:
  7. discriminator = Conv2D(filters = 64, kernel_size = (4,4), strides = (2,2), padding = "same", data_format = "channels_last", kernel_initializer = kernel_init)(dis_input)
  8. discriminator = LeakyReLU(0.2)(discriminator)
  9. # Conv layer 2:
  10. discriminator = Conv2D(filters = 128, kernel_size = (4,4), strides = (2,2), padding = "same", data_format = "channels_last", kernel_initializer = kernel_init)(discriminator)
  11. discriminator = BatchNormalization(momentum = 0.5)(discriminator)
  12. discriminator = LeakyReLU(0.2)(discriminator)
  13. # Conv layer 3:
  14. discriminator = Conv2D(filters = 256, kernel_size = (4,4), strides = (2,2), padding = "same", data_format = "channels_last", kernel_initializer = kernel_init)(discriminator)
  15. discriminator = BatchNormalization(momentum = 0.5)(discriminator)
  16. discriminator = LeakyReLU(0.2)(discriminator)
  17. # Conv layer 4:
  18. discriminator = Conv2D(filters = 512, kernel_size = (4,4), strides = (2,2), padding = "same", data_format = "channels_last", kernel_initializer = kernel_init)(discriminator)
  19. discriminator = BatchNormalization(momentum = 0.5)(discriminator)
  20. discriminator = LeakyReLU(0.2)(discriminator)#discriminator = MaxPooling2D(pool_size=(2, 2))(discriminator)
  21. # Flatten
  22. discriminator = Flatten()(discriminator)
  23. # Dense Layer
  24. discriminator = Dense(1)(discriminator)
  25. # Sigmoid Activation
  26. discriminator = Activation('sigmoid')(discriminator)
  27. # Optimizer and Compiling model
  28. dis_opt = Adam(lr=0.0002, beta_1=0.5)
  29. discriminator_model = Model(input = dis_input, output = discriminator)
  30. discriminator_model.compile(loss='binary_crossentropy', optimizer=dis_opt, metrics=['accuracy'])
  31. discriminator_model.summary()
  32. return discriminator_model
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement