Konvertera TensorFlow-modellen till ONNX

I föregående steg i den här självstudien skapade vi en maskininlärningsmodell med TensorFlow. Nu ska vi konvertera den till ONNX-formatet.

Här använder tf2onnx vi verktyget för att konvertera vår modell genom att följa dessa steg.

1. Spara tf-modellen som förberedelse för ONNX-konvertering genom att köra följande kommando.

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

2. Installera tf2onnx och onnxruntime, genom att köra följande kommandon.

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

3. Konvertera modellen genom att köra följande kommando.

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

Nästa steg

Nu har vi konverterat vår modell till ett ONNX-format som är lämpligt för användning med Windows Machine Learning-API:er. I det sista steget i den här självstudien integrerar vi den i en Windows-app.