欢迎访问 生活随笔！
TAG聚合

生活随笔

当前位置： 首页 > 编程资源 > 编程问答 >内容正文

编程问答

将keras的h5模型转化为onnx

发布时间：2025/4/5 编程问答 44 豆豆
生活随笔 收集整理的这篇文章主要介绍了 将keras的h5模型转化为onnx 小编觉得挺不错的,现在分享给大家,帮大家做个参考.

将keras的h5模型转化为onnx
先安装

pip install keras2onnx import keras import keras2onnx import onnx from keras.models import load_model model = load_model('/root/notebook/model/river_model5.h5') onnx_model = keras2onnx.convert_keras(model, model.name) temp_model_file = '/root/notebook/model/model.onnx' onnx.save_model(onnx_model, temp_model_file)

参考官方文档
https://pypi.org/project/keras2onnx/
官方文档是个好东西

link

总结

以上是生活随笔为你收集整理的将keras的h5模型转化为onnx的全部内容，希望文章能够帮你解决所遇到的问题。

如果觉得生活随笔网站内容还不错，欢迎将生活随笔推荐给好友。