728x90 반응형 경량화1 [ONNX] ONNX로 pytorch 딥러닝 모델 변환/추출/실행하기 (Yolov5) 다른 개발자 분과 이야기하다가 ONNX에 대해 접하게 되었습니다. 사실 ONNX를 코드에서 봤을 때는 뭐 새로운 게 딥러닝 라이브러리인가? 싶었는데, 생각보다 활용성이 많은 것 같아, 이번에 한번 간단하게 테스트를 해보고자 합니다. 간단하게 ONNX의 개념을 살펴보고, ONNX변환 코드를 소개하겠습니다. ONNX 란? ONNX는 다른 환경에서 제작한 딥러닝 모델을 호환해서 사용할 수 도록 해주는 교차 플랫폼 모델 가속기입니다. ONNX는 pytorch, tensorflow/keras, tflit, scikit-learn 및 기타 프레임워크 모델과 함께 사용할 수 있습니다. Pytorch에서 작성한 모델을 ONNX로 변환하는 방법 1. onnx 파일 추출 2. onnxruntime 이용하여 실행 onnx .. 2022. 12. 15. 이전 1 다음 728x90 반응형