Advertisement
Guest User

Untitled

a guest
Aug 24th, 2017
70
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.70 KB | None | 0 0
  1. dl_model = Sequential()
  2.  
  3. dl_model.add(Conv2D(32, (3, 3), input_shape=(64, 48, 3), activation='relu'))
  4. dl_model.add(Conv2D(32, (3, 3), activation='relu'))
  5. dl_model.add(MaxPooling2D(pool_size=(2, 2)))
  6.  
  7. dl_model.add(Flatten())
  8. dl_model.add(Dense(128, activation='relu'))
  9. dl_model.add(Dropout(0.5))
  10. dl_model.add(Dense(1, activation='sigmoid'))
  11.  
  12. adam = Adam(lr=0.00001)
  13.  
  14. dl_model.compile(loss='binary_crossentropy', optimizer=adam, metrics=['accuracy'])
  15.  
  16.  
  17. train_data_generator = ImageDataGenerator(rescale=1. / 255,
  18. shear_range=0.2,
  19. zoom_range=0.2,
  20. rotation_range=40,
  21. width_shift_range=0.2,
  22. height_shift_range=0.2,
  23. horizontal_flip=True,
  24. fill_mode='nearest')
  25.  
  26. test_data_generator = ImageDataGenerator(rescale=1. / 255)
  27.  
  28. training_set = train_data_generator.flow_from_directory(path_training,
  29. target_size=(64, 48),
  30. batch_size=32,
  31. class_mode='binary')
  32.  
  33. validation_set = test_data_generator.flow_from_directory(path_validation,
  34. target_size=(64, 48),
  35. batch_size=32,
  36. class_mode='binary')
  37.  
  38.  
  39. dl_model.fit_generator(training_set, 8000, args.epochs, validation_data=validation_set, validation_steps=2000, verbose=1)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement