Převod modelu TensorFlow na ONNX

V předchozím kroku tohoto kurzu jsme vytvořili model strojového učení s TensorFlow. Teď ho převedeme do formátu ONNX.

Zde použijeme nástroj tf2onnx k převedení našeho modelu podle následujících kroků.

1. Uložte model tf v rámci přípravy na převod ONNX spuštěním následujícího příkazu.

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

2. Nainstalujte tf2onnx a onnxruntimespusťte následující příkazy.

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

3. Převeďte model spuštěním následujícího příkazu.

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

Další kroky

Model jsme teď převedli na formát ONNX, který je vhodný pro použití s rozhraními API služby Windows Machine Learning. V poslední fázi tohoto kurzu ho integrujeme do aplikace pro Windows.