딥러닝 모델 아키텍쳐를 그림으로 확인해보기! : NETRON

작성자 : 한양대학원 융합로봇시스템학과 유승환 석사과정 (CAI LAB)

 

 오늘은 딥러닝 모델 아키텍쳐를 그림으로 확인해 볼 수 있는 Visualization Tool인 NETRON을 사용해보겠습니다! 개인적으로 yolov 계열의 모델 아키텍쳐를 그림으로 확인해보고 싶었는데, 이번 기회에 NETRON을 활용하여 yolo 아키텍쳐를 볼 수 있었습니다! 그러면 시작하겠습니다~

---

 링크 1) NETRON 깃헙 링크 :  github.com/lutzroeder/netron

lutzroeder/netron

---

1. 설치법

 설치하는 방법은 간단합니다. window 10 기준, 파란색 Download 버튼을 클릭하여 exe 파일을 다운받습니다. 그리고 일반적인 프로그램 설치법과 동일하게 exe 파일을 실행시켜서 NETRON을 설치합니다. 

---

2. 사용법

 처음에 Netron을 실행시키면 아래와 같은 화면이 나옵니다. 

 위 화면에 우리가 보고 싶은 모델 파일을 드래그해서 올리면 해당 모델의 아키텍쳐가 뜹니다! 정말 쉽죠?! 저와 같이 yolov4-tiny를 보고 싶다면 darknet의 yolov4-tiny.cfg 파일을 드래그해서 netron에 올리면 됩니다~ 그러면 아래와 같이 yolov4-tiny의 아키텍쳐를 netron에서 보여줍니다. netron의 좋은 점은 블록마다 input, output 형태 뿐만 아니라 어떤 activation function을 사용하는지도 보여줍니다. 

yolov4-tiny의 아키텍쳐 (일부, NETRON)

 아쉽게도 pytorch 모델을 보여주는 것은 개발중이라... yolov5의 아키텍쳐는 yolov4와 같이 상세하게 볼 수 없었습니다. 

yolov5s 아키텍쳐 (일부, NETRON)

---

3. 지원하는 딥러닝 라이브러리

 3-1) 공식 지원 라이브러리

---

 3-2) 개발중인 라이브러리 (비공식 지원)

---