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

Question

我最近发现了这个Project，它使用webgl在支持GPU的浏览器中运行keras模型的推理。我有几个tensorflow项目，我想在浏览器上运行推理，有没有办法把tensorflow模型导出到hdf5文件中，这样它就可以用KerasJS运行了

Answer 1

我能看到这个工作的唯一方法是使用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

Answer 2

如果你使用Keras，你可以这样做。在

model.save_weights('my_model.hdf5')