Advertisement
Guest User

Untitled

a guest
Jun 15th, 2019
93
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.50 KB | None | 0 0
  1. def getModelWeightsAsDict():
  2.  
  3. keras.backend.clear_session()
  4. tf.reset_default_graph()
  5. model = keras.applications.VGG19(input_shape=(IMAGE_SIZE,IMAGE_SIZE,3),include_top=False,weights='imagenet')
  6.  
  7. with keras.backend.get_session() as sess:
  8. weights_dict = {}
  9. for i in range(len(model.layers)):
  10. weights_dict['layer_' + str(i)] = []
  11. for j in range(len(model.layers[i].weights)):
  12. weights_dict['layer_' + str(i)].append(sess.run(model.layers[i].weights[j]))
  13. tf.reset_default_graph()
  14.  
  15. return weights_dict
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement