YOLOv8 분류 모델의 ONNX Runtime 추론을 위한 전처리 파이프라인 구현
Ultralytics의 YOLOv8 프레임워크는 분류, 탐지, 분할 등 다양한 비전 작업을 한 줄의 코드로 학습하고 배포할 수 있는 강력한 기능을 제공합니다. 하지만 .pt 가중치를 ONNX 형식으로 내보낸 후, ultralytics 라이브러리가 없는 환경에서 onnxruntime만으로 추론을 시도하면 결과값이 일치하지 않거나 신뢰도(Confidence)가 현저히 낮게 측정되는 경우가 발생합니다.
이 ...
7월 13일 02:29에 게시됨