이 자습서의 이전 단계에서는 TensorFlow를 사용하여 기계 학습 모델을 만들었습니다. 이제 ONNX 형식으로 변환하겠습니다.
여기서는 다음 단계에 따라 이 도구를 사용하여 tf2onnx 모델을 변환합니다.
- 다음 명령을 실행하여 ONNX 변환을 준비하여 tf 모델을 저장합니다.
python save_model.py --weights ./data/yolov4.weights --output ./checkpoints/yolov4.tf --input_size 416 --model yolov4
-
tf2onnx및onnxruntime를 설치하려면 다음 명령을 실행하세요.
pip install onnxruntime
pip install git+https://github.com/onnx/tensorflow-onnx
- 다음 명령을 실행하여 모델을 변환합니다.
python -m tf2onnx.convert --saved-model ./checkpoints/yolov4.tf --output model.onnx --opset 11 --verbose
다음 단계
이제 모델을 Windows Machine Learning API에 사용하기에 적합한 ONNX 형식으로 변환했습니다. 이 자습서의 마지막 단계에서는 Windows 앱에 통합하는 작업을 수행합니다.