將keras的h5模型轉(zhuǎn)化為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

總結(jié)

以上是生活随笔為你收集整理的将keras的h5模型转化为onnx的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。