Barracuda :: NETRON 사용해서 ONNX모델 구조 확인 하는 법

Netron은 기계 학습(Machine Learning) 모델을 시각화하기 위한 도구로, 다양한 딥러닝 프레임워크에서 생성된 모델 파일을 쉽게 열어보고 이해할 수 있게 도와줍니다. 이 도구는 사용자 친화적 GUI를 제공하며, 복잡한 신경망 구조를 그래프로 시각화하여 모델의 아키텍처를 명확히 파악할 수 있도록 설계되었습니다.

 

https://github.com/lutzroeder/netron

GitHub - lutzroeder/netron: Visualizer for neural network, deep learning and machine learning models

 

사용 방법

1. 설치
   • Netron은 데스크톱 앱(Windows, macOS, Linux)으로 설치하거나 브라우저 기반 웹 버전으로 사용 가능합니다.
   • 데스크톱 버전 설치:
     • Netron 공식 웹사이트에서 다운로드 후 설치.
2. 모델 열기
   • Netron을 실행한 후, 모델 파일(.onnx, .pb 등)을 드래그하여 창에 놓으면 모델 구조가 자동으로 로드됩니다.
3. 탐색
   • 각 레이어를 클릭하여 세부 정보를 확인하거나, 줌 인/줌 아웃으로 모델의 전체 구조를 탐색할 수 있습니다.

 

설치하고 나서 onnx 파일을 열면 다음과 같이 확인이 가능합니다.

 

 

유니티 안에서 onnx 결과 확인하는 방법으로

onnx 모델을 선택하고 Open Imported NN model as temp file 버튼을 클릭해주면

모델을 실행한 결과를 확인 할 수 있습니다

 

왼쪽 : Exported Graph(onnx)/ 오른쪽 : Imported Graph(Barracuda)

 

바라쿠다는 임포트시에 최적화를 하기 때문에 이런식으로 내부 표현이 달라짐