Guest User

Untitled

a guest
Mar 21st, 2018
80
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.76 KB | None | 0 0
  1. model = models.Sequential()
  2. model.add(layers.Conv2D(32, (3, 3), activation='relu',
  3. input_shape=(1500, 1500, 3)))
  4. model.add(layers.MaxPooling2D((2, 2)))
  5. model.add(layers.Conv2D(64, (3, 3), activation='relu'))
  6. model.add(layers.MaxPooling2D((2, 2)))
  7. model.add(layers.Conv2D(128, (3, 3), activation='relu'))
  8. model.add(layers.MaxPooling2D((2, 2)))
  9. model.add(layers.Conv2D(128, (3, 3), activation='relu'))
  10. model.add(layers.MaxPooling2D((2, 2)))
  11. model.add(layers.Conv2D(256, (3, 3), activation='relu'))
  12. model.add(layers.MaxPooling2D((2, 2)))
  13. model.add(layers.Conv2D(256, (3, 3), activation='relu'))
  14. model.add(layers.MaxPooling2D((2, 2)))
  15. model.add(layers.Flatten())
  16. model.add(layers.Dense(512, activation='relu'))
  17. model.add(layers.Dense(1, activation='sigmoid'))
  18.  
  19. # All images will be rescaled by 1./255
  20. train_datagen = ImageDataGenerator(rescale=1./255)
  21. test_datagen = ImageDataGenerator(rescale=1./255)
  22.  
  23. train_generator = train_datagen.flow_from_directory(
  24. # This is the target directory
  25. train_dir,
  26. # All images will be resized to 1500x1500
  27. target_size=(1500, 1500),
  28. batch_size=2,
  29. # Since we use binary_crossentropy loss, we need binary labels
  30. class_mode='binary')
  31.  
  32. validation_generator = test_datagen.flow_from_directory(
  33. validation_dir,
  34. target_size=(1500, 1500),
  35. batch_size=2,
  36. class_mode='binary')
  37.  
  38. test_generator = test_datagen.flow_from_directory(
  39. test_dir,
  40. target_size=(1500, 1500),
  41. batch_size=2,
  42. class_mode='binary')
  43.  
  44. probabilities = model.predict_generator(test_generator,5)
  45.  
  46. array([[2.6628117e-28],
  47. [6.6314442e-06],
  48. [3.2372427e-20],
  49. [7.8302348e-04],
  50. [1.0000000e+00],
  51. [1.0000000e+00],
  52. [1.0000000e+00],
  53. [8.4590050e-14],
  54. [9.9938679e-01],
  55. [3.6370282e-25]], dtype=float32)
Add Comment
Please, Sign In to add comment