TensorFlow-model converteren naar ONNX

In de vorige stap van deze zelfstudie hebben we een machine learning-model gemaakt met TensorFlow. Nu converteren we deze naar de ONNX-indeling.

Hier gebruiken we het tf2onnx hulpprogramma om ons model te converteren. Volg deze stappen.

1. Sla het tf-model op in voorbereiding op ONNX-conversie door de volgende opdracht uit te voeren.

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

2. Installeer tf2onnx en onnxruntimevoer de volgende opdrachten uit.

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

3. Converteer het model door de volgende opdracht uit te voeren.

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

Volgende stappen

We hebben ons model nu geconverteerd naar een ONNX-indeling, geschikt voor gebruik met Windows Machine Learning-API's. In de laatste fase van deze zelfstudie integreren we deze in een Windows-app.