Guest User

Untitled

a guest
May 20th, 2018
143
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.78 KB | None | 0 0
  1. base = VGG16(weights='imagenet', include_top=False, input_shape=(64,64,3))
  2.  
  3. # convert to sequential model
  4. model = Sequential()
  5. for layer in base.layers:
  6. model.add(layer)
  7.  
  8. # Remove last layer
  9. model.layers.pop()
  10.  
  11. # add flatten and two dense layers that don't appear when specifying an input_shape
  12. model.add(Flatten())
  13. model.add(Dense(4096))
  14. model.add(Dropout(.5))
  15. model.add(Dense(4096))
  16. model.add(Dropout(.5))
  17.  
  18. for layer in model.layers:
  19. layer.trainable = False
  20.  
  21. # Add a layer for 3 classes
  22. model.add(Dense(3, activation='softmax'))
  23.  
  24. model.compile(
  25. optimizer = 'rmsprop',
  26. loss='categorical_crossentropy',
  27. metrics=['accuracy']
  28. )
  29.  
  30. # training
  31. model.fit(x_train,y_train, epochs=30, batch_size=64, verbose=1)
  32.  
  33. # predict
  34. y_target = model.predict(x_target, batch_size=64, verbose=1)
Add Comment
Please, Sign In to add comment