Notion百科机器学习
Source
Edit
History

Tensorflow kears模型转ONNX 格式.md

Catalogue
  1. 1. Tensorflow kears模型转ONNX 格式
    1. 1.1. 方案一 （不适用于我的情况，不排除其可行性）
    2. 1.2. 方案二

Tensorflow kears模型转ONNX 格式

环境 ： ubuntu 20.04 TensorFlow2.4 Pycharm

需要将kears模型转换为onnx格式,以供OpenCV调用

方案一 （不适用于我的情况，不排除其可行性）

1
2
3
4
5
6
7
8
9
import keras
import keras2onnx
import onnx
from tensorflow.keras.models import load_model
model = load_model('REDPropiaFinal.h5')
onnx_model = keras2onnx.convert_keras(model, model.name)
temp_model_file = 'REDPropiaFinal.onnx'
onnx.save_model(onnx_model, temp_model_file)

方案二

1
2
3
4
5
6
7
8
9
10
11
# Install helper packages:
!pip install tf2onnx onnx onnxruntime

# Load model from .h5 and save as Saved Model:
import tensorflow as tf
model = tf.keras.models.load_model("REDPropiaFinal.h5")
tf.saved_model.save(model, "tmp_model")

# Convert in bash:
!python -m tf2onnx.convert --saved-model tmp_model --output "REDPropiaFinal.onnx"