TensorFlow 모델을 ONNX로 변환

이 자습서의 이전 단계에서는 TensorFlow를 사용하여 기계 학습 모델을 만들었습니다. 이제 이를 ONNX 형식으로 변환합니다.

여기서는 다음 단계에 따라 tf2onnx 도구를 사용하여 모델을 변환합니다.

1. 다음 명령을 실행하여 ONNX 변환 준비 중에 tf 모델을 저장합니다.

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 형식으로 변환했으며 Windows Machine Learning API에 사용하기에 적합합니다. 이 자습서의 마지막 단계에서는 Windows 앱에 통합합니다.