Guest User

Untitled

a guest
Nov 14th, 2018
125
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.19 KB | None | 0 0
  1. train_datagen = ImageDataGenerator(rescale=1./255, zoom_range=0.3, rotation_range=50,
  2. width_shift_range=0.2, height_shift_range=0.2, shear_range=0.2,
  3. horizontal_flip=True, fill_mode='nearest')
  4. val_datagen = ImageDataGenerator(rescale=1./255)
  5. train_generator = train_datagen.flow(train_imgs, train_labels_enc, batch_size=30)
  6. val_generator = val_datagen.flow(validation_imgs, validation_labels_enc, batch_size=20)
  7.  
  8. from keras.layers import Conv2D, MaxPooling2D, Flatten, Dense, Dropout, InputLayer
  9. from keras.models import Sequential
  10. from keras import optimizers
  11.  
  12. model = Sequential()
  13. model.add(vgg_model)
  14. model.add(Dense(512, activation='relu', input_dim=input_shape))
  15. model.add(Dropout(0.3))
  16. model.add(Dense(512, activation='relu'))
  17. model.add(Dropout(0.3))
  18. model.add(Dense(1, activation='sigmoid'))
  19.  
  20. model.compile(loss='binary_crossentropy',
  21. optimizer=optimizers.RMSprop(lr=1e-5),
  22. metrics=['accuracy'])
  23.  
  24. history = model.fit_generator(train_generator, steps_per_epoch=100, epochs=100,
  25. validation_data=val_generator, validation_steps=50,
  26. verbose=1)
Add Comment
Please, Sign In to add comment