將張量流權重輸出到hdf5文件和模型到keras model.json

我最近發現這個Project，它在具有使用webgl的GPU支持的瀏覽器中運行keras模型的推理。我有幾個tensorflow項目，我想在瀏覽器上運行推理，有沒有辦法將tensorflow模型導出到hdf5文件中，以便可以使用keras -js運行將張量流權重輸出到hdf5文件和模型到keras model.json

來源

2017-04-13 anandaravindan

您可以檢查這個問題我的回答另一個線程。 https://stackoverflow.com/questions/44466066/how-can-i-convert-a-trained-tensorflow-model-to-keras/46210187#46210187 – user1190882

如果您使用Keras，您可以執行像這樣的東西。

model.save_weights('my_model.hdf5')

來源

2017-04-14 06:07:40

不，我正在使用張量流 – anandaravindan

我可以看到這個工作的唯一方法是，如果你使用一個Keras模型爲您的工作流程TensorFlow的接口。如果你這樣做，你可以做到這一點，以節省模型及其權重：

# 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)

使用Keras爲Tensorflow接口的更多信息工作流程在這裏：https://blog.keras.io/keras-as-a-simplified-interface-to-tensorflow-tutorial.html#using-keras-models-with-tensorflow

來源

2017-04-14 20:18:19

將張量流權重輸出到hdf5文件和模型到keras model.json

回答

相關問題