[ONNX] ONNX Model Visualization(Netron)

이번 포스팅에서는 ONNX 모델을 시각화 할 수 있는 Netron에 대해서 설명하겠습니다!

 

Github: https://github.com/lutzroeder/netron

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

 

Browser: https://netron.app/

Netron

 

다운로드하여 PC에 직접 설치해 사용할 수도 있고 별도의 설치 없이 웹사이트에 ONNX 모델을 바로 업로드 해 확인해 볼 수도 있습니다.

또한, 아래 그림처럼 구조도 뿐만 아니라 layer를 클릭하면 학습된 파라미터 값도 확인해 볼 수 있습니다.

 

Tensorflow Lite, Tensorflow, Keras 등도 사용 가능합니다. (.pth 파일도 사용 가능 하지만 onnx 모델만큼 예쁘게 그려지지는 않네요!)