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

Question

我最近發現了這個項目 ，該項目使用webgl在具有GPU支持的瀏覽器中運行keras模型的推斷。 我有一些tensorflow項目，我想在瀏覽器上運行推理，有沒有一種方法可以將tensorflow模型導出到hdf5文件中，以便可以使用keras-js運行

Answer 1

如果您使用Keras，則可以執行以下操作。

model.save_weights('my_model.hdf5')

Answer 2

我看到此工作的唯一方法是，如果您使用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