Konvertieren des TensorFlow-Modells in ONNX

Im vorherigen Schritt dieses Lernprogramms haben wir ein Machine Learning-Modell mit TensorFlow erstellt. Jetzt konvertieren wir sie in das ONNX-Format.

Hier verwenden wir das tf2onnx Tool, um unser Modell zu konvertieren. Führen Sie dazu die folgenden Schritte aus.

1. Speichern Sie das TF-Modell in Vorbereitung auf die ONNX-Konvertierung, indem Sie den folgenden Befehl ausführen.

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

2. Installieren tf2onnx und onnxruntime, indem Sie die folgenden Befehle ausführen.

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

3. Konvertieren Sie das Modell, indem Sie den folgenden Befehl ausführen.

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

Nächste Schritte

Wir haben nun unser Modell in ein ONNX-Format konvertiert, das für die Verwendung mit Windows Machine Learning-APIs geeignet ist. In der letzten Phase dieses Lernprogramms integrieren wir es in eine Windows-App.