TensorFlow-modell átalakítása ONNX-gé

Az oktatóanyag előző lépésében létrehoztunk egy gépi tanulási modellt a TensorFlow használatával. Most az ONNX formátumot fogjuk átalakítani.

Itt az tf2onnx eszközt használjuk a modellünk átalakítására a következő lépések szerint.

1. Mentse a tf-modellt az ONNX-átalakítás előkészítése során az alábbi parancs futtatásával.

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

2. Telepítse a tf2onnx és a onnxruntime, majd futtassa a következő parancsokat.

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

3. Konvertálja a modellt az alábbi parancs futtatásával.

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

Következő lépések

Most egy ONNX formátummá alakítottuk át a modellt, amely a Windows Machine Learning API-khoz használható. Az oktatóanyag utolsó szakaszában integráljuk azt egy Windows-alkalmazásba.