将已保存的.h5文件转换为JSON文件

9

我使用了Keras中的预制模型来训练关于URL的数据集,结果已经保存为.h5格式。我想知道是否可以将保存的.h5文件转换成JSON格式,这样我就可以查看结果并将JSON文件应用到我的系统中,从而将坏的URL与好的URL分开。如果可能的话,我也希望避免重新进行训练。


你可以在这里参考答案:https://dev59.com/MF4c5IYBdhLWcg3wOoEH - Han Wang
我需要将它转换为JSON格式,这样我就可以使用JavaScript调用该JSON文件并从中获取数据。 - Alexander Wazowski
你可以参考这里的答案将数据保存为json格式:https://dev59.com/vmct5IYBdhLWcg3wI6N6 - Manoj Mohan
2个回答

3
如同在https://www.tensorflow.org/js/guide/conversion中所提到的一样。
例如,假设您已将名为model.h5的Keras模型保存在tmp/目录中。要使用TensorFlow.js转换器转换您的模型,您可以运行以下命令:
tensorflowjs_converter --input_format=keras /tmp/model.h5 /tmp/tfjs_model

0

首先

!sudo pip3 install tensorflowjs

那么

!sudo tensorflowjs_converter

只需将model.h5替换为您训练好的权重即可

!tensorflowjs_converter --input_format=keras '/content/drive/MyDrive/model.h5' "/content/drive/MyDrive/json"

你已经准备就绪,你的文件将被转换!


网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接