Mengonversi model TensorFlow ke ONNX

Pada langkah sebelumnya dari tutorial ini, kami membuat model pembelajaran mesin dengan TensorFlow. Sekarang, kita akan mengonversinya ke format ONNX.

Di sini, kita akan menggunakan tf2onnx alat untuk mengonversi model kita, mengikuti langkah-langkah ini.

1. Simpan model tf sebagai persiapan untuk konversi ONNX, dengan menjalankan perintah berikut.

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

2. Instal tf2onnx dan onnxruntime, dengan menjalankan perintah berikut.

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

3. Konversi model, dengan menjalankan perintah berikut.

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

Langkah selanjutnya

Kami sekarang telah mengonversi model kami ke format ONNX, cocok untuk digunakan dengan API Pembelajaran Mesin Windows. Pada tahap akhir tutorial ini, kami mengintegrasikannya ke dalam aplikasi Windows.