将tensorflow权重导出到hdf5文件并将模型导出到keras model.json [英] Export tensorflow weights to hdf5 file and model to keras model.json

问题描述

我最近发现了这个项目，该项目在具有GPU支持的浏览器中运行keras模型的推断使用webgl.我有一些tensorflow项目，我想在浏览器上运行推理，有没有一种方法可以将tensorflow模型导出到hdf5文件中，以便可以使用keras-js来运行

I recently found this Project which runs inference of keras model in a browser with GPU support using webgl. I have a few tensorflow project that I would like to run inference on a browser, is there a way to export tensorflow models into hdf5 file so it can be run using keras-js

推荐答案

如果您使用Keras，则可以执行以下操作.

If you are using Keras, you can do something like this.

model.save_weights('my_model.hdf5')

这篇关于将tensorflow权重导出到hdf5文件并将模型导出到keras model.json的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！