2017-04-13 1 views
0

J'ai récemment trouvé ce Project qui exécute l'inférence du modèle keras dans un navigateur avec le soutien du GPU en utilisant webgl. J'ai quelques projets tensorflow que je voudrais faire inference sur un navigateur, est-il un moyen d'exporter des modèles tensorflow dans le fichier hdf5 afin qu'il puisse être exécuté en utilisant keras-jsExport poids tensorflow en fichier hdf5 et le modèle à keras model.json

+0

Vous pouvez vérifier ma réponse sur cette question dans un autre fil. https://stackoverflow.com/questions/44466066/how-can-i-convert-a-trained-tensorflow-model-to-keras/46210187#46210187 – user1190882

Répondre

2

Si vous utilisez Keras, vous pouvez faire quelque chose comme ça.

model.save_weights('my_model.hdf5') 
+0

Non J'utilise tensorflow – anandaravindan

0

La seule façon que je peux voir ce travail est si vous utilisez un modèle Keras comme interface à votre flux de travail TensorFlow. Si vous faites cela, vous pouvez le faire pour sauver le modèle et ses poids:

# save model 
with open(model_save_filename, "w") as model_save_file: 
    model_json = model.to_json() 
    model_save_file.write(model_json) 

# save model weights 
model.save_weights(model_weights_save_filename) 

Plus d'informations sur l'utilisation Keras comme interface Tensorflow workflows ici: https://blog.keras.io/keras-as-a-simplified-interface-to-tensorflow-tutorial.html#using-keras-models-with-tensorflow