通过

将 TensorFlow 模型转换为 ONNX

在 本教程的上一步中，我们使用 TensorFlow 创建了机器学习模型。 现在，我们将将其转换为 ONNX 格式。

在这里，我们将使用该工具 tf2onnx 来转换我们的模型，按照以下步骤进行。

  1. 运行以下命令，保存 tf 模型以准备 ONNX 转换。

python save_model.py --weights ./data/yolov4.weights --output ./checkpoints/yolov4.tf --input_size 416 --model yolov4

  1. tf2onnx和onnxruntime，通过运行以下命令进行安装。
pip install onnxruntime
pip install git+https://github.com/onnx/tensorflow-onnx
  1. 通过运行以下命令转换模型。

python -m tf2onnx.convert --saved-model ./checkpoints/yolov4.tf --output model.onnx --opset 11 --verbose

后续步骤

现在，我们已将模型转换为 ONNX 格式，适合用于 Windows 机器学习 API。 在本教程的最后阶段，我们将 它集成到 Windows 应用中。