[tf, keras] 모델 시각화 (netron)

netron

netron github에서 각자의 OS에 맞게 다운 받아 설치하면 됩니다.

 

model 을 학습시키면 보통 결과를 .h5 확장자로 저장합니다.

netron은 .h5를 시각화 해주는 툴입니다. 링크에서 샘플을 실습해 볼 수 있도록 샘플이 올려져 있으니 참고하시면 됩니다.

 

netron을 실행한 후 .h5 모델을 불러오거나 시각화 하고 싶은 .h5 파일을 실행하면 다음과 같이 결과를 볼 수 있습니다.

 

Ref

- 케라스 모델 시각화 https://gaussian37.github.io/dl-keras-%EC%BC%80%EB%9D%BC%EC%8A%A4-%EB%AA%A8%EB%8D%B8-%EC%8B%9C%EA%B0%81%ED%99%94/