將 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

2. 執行下列命令來安裝 tf2onnx 和 onnxruntime 。

pip install onnxruntime
pip install git+https://github.com/onnx/tensorflow-onnx

3. 執行下列命令來轉換模型。

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

下一步

我們現在已將模型轉換成 ONNX 格式，適用于 Windows 機器學習 API。 在本教學課程的最後階段，我們會 將其整合到 Windows 應用程式中 。