오늘 포스팅에서는 onnx 모델을 사용할 때 유용한 기능을 가지고 있는 Netron에 대해 다뤄보겠습니다.
모델 구조를 파악하고, 시각화 할 때 사용해보아요 ᡣ • . • 𐭩 ♡
Netron 설치하기
리눅스 환경에서는 간단한 apt 설치를 통해 사용할 수 있습니다.
sudo apt install netron
# sudo snap install netron
Netron 사용하기
모델이 있는 경로에서 터미널창을 띄우고, 아래 명령어를 입력해줍니다.
netron tour_model.onnx
그럼 아래와 같은 메세지가 나오는데, http~에 해당하는 주소로 접속을 하시면 됩니다.
이처럼 모델 구조가 시각화되어 전체적인 구조를 확인할 수 있습니다.
이미지를 저장하려면 좌측 상단의 메뉴 아이콘을 클릭하시고 저장하시면 됩니다.
그 외에도 모델 구조를 하나하나 클릭해보면 파라미터 값들도 확인할 수 있습니다.
