Guest User

Untitled

a guest
Nov 14th, 2018
166
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.33 KB | None | 0 0
  1. train_generator = train_datagen.flow(train_imgs, train_labels_enc, batch_size=30)
  2. val_generator = val_datagen.flow(validation_imgs, validation_labels_enc, batch_size=20)
  3. input_shape = (150, 150, 3)
  4.  
  5. from keras.layers import Conv2D, MaxPooling2D, Flatten, Dense, Dropout
  6. from keras.models import Sequential
  7. from keras import optimizers
  8.  
  9. model = Sequential()
  10.  
  11. model.add(Conv2D(16, kernel_size=(3, 3), activation='relu',
  12. input_shape=input_shape))
  13. model.add(MaxPooling2D(pool_size=(2, 2)))
  14.  
  15. model.add(Conv2D(64, kernel_size=(3, 3), activation='relu'))
  16. model.add(MaxPooling2D(pool_size=(2, 2)))
  17.  
  18. model.add(Conv2D(128, kernel_size=(3, 3), activation='relu'))
  19. model.add(MaxPooling2D(pool_size=(2, 2)))
  20.  
  21. model.add(Conv2D(128, kernel_size=(3, 3), activation='relu'))
  22. model.add(MaxPooling2D(pool_size=(2, 2)))
  23.  
  24. model.add(Flatten())
  25. model.add(Dense(512, activation='relu'))
  26. model.add(Dropout(0.3))
  27. model.add(Dense(512, activation='relu'))
  28. model.add(Dropout(0.3))
  29. model.add(Dense(1, activation='sigmoid'))
  30.  
  31. model.compile(loss='binary_crossentropy',
  32. optimizer=optimizers.RMSprop(lr=1e-4),
  33. metrics=['accuracy'])
  34.  
  35. history = model.fit_generator(train_generator, steps_per_epoch=100, epochs=100,
  36. validation_data=val_generator, validation_steps=50,
  37. verbose=1)
Add Comment
Please, Sign In to add comment