Advertisement
Guest User

Untitled

a guest
Apr 21st, 2019
97
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.43 KB | None | 0 0
  1. from keras.models import Sequential
  2. from keras.layers import Dense
  3. from keras.layers import Flatten
  4. from keras.layers import Convolution2D
  5. from keras.layers import MaxPooling2D
  6.  
  7. classifier = Sequential()
  8.  
  9. classifier.add(Convolution2D(32,3,3,input_shape=(64,64,3),activation = 'relu'))
  10. classifier.add(MaxPooling2D(pool_size = (2,2)))
  11. classifier.add(Flatten())
  12. classifier.add(Dense(output_dim = 128,activation = 'relu' ))
  13. classifier.add(Dense(output_dim = 1 ,activation = 'sigmoid'))
  14. classifier.compile(optimizer = 'adam',loss='binary_crossentropy',metrics = ['accuracy'])
  15. from keras.preprocessing.image import ImageDataGenerator
  16. train_datagen = ImageDataGenerator(
  17. rescale=1./255,
  18. shear_range = 0.2,
  19. zoom_range = 0.2,
  20. horizontal_flip = True)
  21. test_datagen = ImageDataGenerator(rescale = 1./255)
  22. training_set = train_datagen.flow_from_directory(
  23. 'dogscats/train',
  24. target_size= (64,64),
  25. batch_size=32,
  26. shuffle = True,
  27. class_mode = 'binary')
  28. test_set = test_datagen.flow_from_directory(
  29. 'dogscats/test1',
  30. target_size= (64,64),
  31. batch_size=32,
  32. shuffle = True,
  33. class_mode = 'binary')
  34. from IPython.display import display
  35. from PIL import Image
  36.  
  37. classifier.fit_generator(
  38. training_set,
  39. steps_per_epoch=8000,
  40. epochs=10,
  41. validation_data=test_set,
  42. validation_steps=800)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement