728x90
반응형
이번 포스팅에서는 ONNX 모델을 시각화 할 수 있는 Netron에 대해서 설명하겠습니다!
Github: https://github.com/lutzroeder/netron
GitHub - lutzroeder/netron: Visualizer for neural network, deep learning, and machine learning models
Visualizer for neural network, deep learning, and machine learning models - GitHub - lutzroeder/netron: Visualizer for neural network, deep learning, and machine learning models
github.com
Browser: https://netron.app/
Netron
netron.app
다운로드하여 PC에 직접 설치해 사용할 수도 있고 별도의 설치 없이 웹사이트에 ONNX 모델을 바로 업로드 해 확인해 볼 수도 있습니다.
또한, 아래 그림처럼 구조도 뿐만 아니라 layer를 클릭하면 학습된 파라미터 값도 확인해 볼 수 있습니다.
Tensorflow Lite, Tensorflow, Keras 등도 사용 가능합니다. (.pth 파일도 사용 가능 하지만 onnx 모델만큼 예쁘게 그려지지는 않네요!)
728x90
반응형
'ONNX' 카테고리의 다른 글
| [ONNX] ONNX Model Quantization (1) | 2023.09.05 |
|---|---|
| [ONNX] ONNX 변환모델에 메타데이터 추가하기 (0) | 2023.04.01 |
| [ONNX] ONNX Runtime에서 실행하기 (0) | 2023.03.31 |
| [ONNX] Pytorch 모델을 ONNX 모델로 변환하기 (0) | 2023.03.30 |