Преобразование модели TensorFlow в ONNX

На предыдущем шаге этого руководства мы создали модель машинного обучения с tensorFlow. Теперь мы преобразуем его в формат ONNX.

Здесь мы будем использовать tf2onnx средство для преобразования модели, выполнив следующие действия.

1. Сохраните модель tf в подготовке к преобразованию ONNX, выполнив следующую команду.

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

2. Установите tf2onnx и onnxruntimeвыполните следующие команды.

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

3. Преобразуйте модель, выполнив следующую команду.

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

Дальнейшие шаги

Теперь мы преобразовали модель в формат ONNX, подходящий для использования с API машинного обучения Windows. На последнем этапе этого руководства мы интегрируем его в приложение Для Windows.