TensorFlow modelini ONNX'e dönüştürme

Bu öğreticinin önceki adımında TensorFlow ile bir makine öğrenmesi modeli oluşturduk. Şimdi bunu ONNX biçimine dönüştüreceğiz.

Burada, tf2onnx aracını kullanarak bu adımları izleyip modelimizi dönüştüreceğiz.

1. Aşağıdaki komutu çalıştırarak TF modelini ONNX'e dönüştürmek için hazırlık aşamasında kaydedin.

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

2. tf2onnx ve onnxruntime yükleyin, ardından aşağıdaki komutları çalıştırın.

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

3. Aşağıdaki komutu çalıştırarak modeli dönüştürün.

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

Sonraki Adımlar

Şimdi modelimizi Windows Machine Learning API'leriyle kullanıma uygun bir ONNX biçimine dönüştürdük. Bu öğreticinin son aşamasında bunu bir Windows uygulamasıyla tümleştiririz.