[ML]Netron(딥러닝 모델 시각화 도구)

Q.딥러닝을 배우면 모델 설계를 하게 되는데 이걸 직관적으로 표현하는 방법은 무엇?

보통 이런 의문이 든다.

필자가 딥러닝이라는 것을 배울때는 아래와 같이 텐서플로, 케라스의 모델 시각화(plot_model)기능으로 시각화를 했었다.

Keras 모델 아키텍처 시각화

 

간단하고 직관적으로 표현되었다는 느낌이 들었고 학부생 과제를 할 때 유용하게 사용했다.

처음에 들어가는 데이터가 어떤 Shape이어야하는지를 파악할 때 편했고 각 레이어를 통과하면서 어떤 Shape으로 변화하는지 잘 알 수 있었다.

 

하지만 필자는 회사를 다니게 되면서 조금더 윗 분들은 복잡한 내용보다는 조금더 직관적이고 이쁘고 알기 쉽게 시각화하기를 원하는 경우가 많았고 이를 위해 더 많은 자료를 찾던중 Netron이라는 것을 발견하게 되었다.

 

Q. Netron은 무엇인가? 그리고 사용법은?

 ▶  Netron은 Tensorflow, Keras, Pytorch, Onnx 등 딥러닝 프레임워크의 모델 아키텍처 시각화 뷰어 역할을 하는 툴이다.

 

▶ 사용법은 간단하다.

각 딥러닝 프레임워크의 모델 저장 기능으로 모델을 로컬 혹은 서버에 저장한다.

웹으로 사용하려면 아래의 페이지를 들어가서 그 모델을 업로드하면 모델이 시각화 되어 나온다.

Netron

 

로컬에서 사용하려면 아래의 깃허브 페이지에 들어가서 각자 os에 맞게 설치하고 웹과 동일하게 각 프레임워크를 모델 저장기능을 통해 모델저장 후 그 모델을 업로드 한다. 

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

 

 

Q. Netron시각화?

필자가 시각화한 LSTM 모델

 

💡 Netron을 알게된 후로 필자는 간단한 딥러닝 아키텍처를 시각화 할 때는 위와 같은 방법으로 시각화하여 보고서를 작성

     한다.

     각 프레임 워크에서 제공하는 시각화 기능도 괜찮지만 필자는 Netron이 조금 더 활용하기 좋았다.

 

💡 모델 아키텍처가 깊어지면 사용하기 어려운면도 있다고 판단되었다.

    그럴때는 간단한 도식화를 사용해야겠다고 판단했다.

    실제로 업무하면서 대규모 모델을 사용할 경우 간단하게 도식화하는 방법으로 보고서를 작성한다.