将tensorflow权重导出到hdf5文件并将模型导出到keras model.json

Question

我最近发现这个Project在使用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