본문 바로가기

onnx

(2)
[Flutter] 플러터에서 ONNX 모델 사용하기 이 포스트에서는 플러터 앱에서 ONNX 모델을 사용할 수 있게 해주는 onnxruntime을 어떻게 사용하는지 알아본다. ONNX에 대해서는 여기를 참고하자.ONNXRUNTIMEpub.dev에는 플러터에서도 ONNX 모델을 사용할 수 있는 onnxruntime 라이브러리를 제공하고 있다. 플러터 앱 개발자는 onnxruntime을 사용하여 원하는 ML 모델을 탑재하여 추론을 할 수 있다. 플러터의 onnxruntime에서 지원하는 OS는 다음과 같다. PlatformAndroidiOSLinuxmacOSWindowsCompatibilityAPI level 21+****Architecturearm32/arm64**** 어지간한 환경에서는 구동된다.XGBoost 모델 추론해 보기ONNX 모델을 구동하기 위해 ..
ONNX in python - XGBoost 모델을 onnx로 저장해보기 ONNX가 뭔데? ONNX는 Open Neural Network eXchange의 약자로, 다양한 환경에서 ML/DL 모델을 구동할 수 있게 해주는 프레임워크이다. 쉽게 말해 인공지능 모델을 크로스 플랫폼으로 사용할 수 있게 해주는 도구다. 이 포스트에서는 훈련한 XGBoost 모델을 onnx 파일로 export 하는 예제 코드를 보여주도록 한다.이 예제는 XGBoost만 된다.왜 XGBoost라고 한정했냐면, 훈련한 모델에 따라 export 방법이 다르기 때문이다. 이 포스트에서는 onnxmltools라는 것을 사용하지만, scikit-learn에 기반하여 훈련된 모델은 skl2onnx라는 모듈을 써서 export 한다. 혹시 파이썬에서 onnx를 쓸 예정이라면 유의하자.예시 코드와 주의점이 코드는 실..