Convertire il modello TensorFlow in ONNX

Nel passaggio precedente di questa esercitazione è stato creato un modello di Machine Learning con TensorFlow. A questo punto, lo convertiremo nel formato ONNX.

In questo caso si userà lo strumento per convertire il tf2onnx modello, seguendo questa procedura.

1. Salvare il modello tf in preparazione per la conversione ONNX eseguendo il comando seguente.

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

2. Installare tf2onnx e onnxruntime, eseguendo i comandi seguenti.

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

3. Convertire il modello eseguendo il comando seguente.

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

Passaggi successivi

Il modello è stato convertito in un formato ONNX, adatto per l'uso con le API di Windows Machine Learning. Nella fase finale di questa esercitazione, la si integra in un'app di Windows.