preloader
Deep learning

Netron! Network 구조를 한번 볼까요?

Netron! Network 구조를 한번 볼까요?

오늘은 netron 이라는 Tool를 드리려 합니다.

본 포스팅은 https://github.com/lutzroeder/netron 내용을 이용하여 작성하였습니다.

간단히 말씀드리면 Neural network viewer 입니다.

굉장히 많은 Framework들을 지원하고 있습니다.

  • ONNX (.onnx.pb.pbtxt)
  • Keras (.h5.keras)
  • Core ML (.mlmodel)
  • Caffe (.caffemodel.prototxt)
  • Caffe2 (predict_net.pb)
  • Darknet (.cfg)
  • MXNet (.model-symbol.json)
  • Barracuda (.nn)
  • ncnn (.param)
  • Tengine (.tmfile)
  • TNN (.tnnproto)
  • UFF (.uff)
  • TensorFlow Lite (.tflite)

이 외에도 불안정하지만 다음과 같은 Framework 도 지원합니다.

  • TorchScript (.pt.pth)
  • PyTorch (.pt.pth)
  • Torch (.t7)
  • Arm NN (.armnn)
  • BigDL (.bigdl.model)
  • Chainer (.npz.h5)
  • CNTK (.model.cntk)
  • Deeplearning4j (.zip)
  • MediaPipe (.pbtxt)
  • ML.NET (.zip)
  • MNN (.mnn)
  • PaddlePaddle (.zip__model__)
  • OpenVINO (.xml)
  • scikit-learn (.pkl)
  • TensorFlow.js (model.json.pb)
  • TensorFlow (.pb.meta.pbtxt.ckpt.index)

(솔직히 Torch, TensorFlow 종류는 왜 나눠서 설명하는지 잘 모르겠음…..하나로 합쳐 놓으면 안되나..)

설치 및 사용법

netron은 PC에 설치해서 사용하거나 설치없이 Browser 버전으로 사용할 수도 있습니다.

  • macOSDownload the .dmg file or run brew cask install netron
  • LinuxDownload the .AppImage file or run snap install netron
  • WindowsDownload the .exe installer or run winget install netron
  • Python Server: Run pip install netron and netron [FILE] or import netron; netron.start('[FILE]')
  • BrowserStart the browser version.

한번 브라우저 버전을 실행시켜 보았습니다.

Open Model 을 클릭 하신 후 보고 싶은 Network 저장 파일을 선택합니다.

Network 구조가 차례로 보이네요.

다음은 keras로 작성된 MobileNetV2 를 netron으로 띄운 후 앞 부분을 잘라낸 사진입니다.

왼쪽에는 네트워크 구조를 볼 수 있고 만약 레이어를 선택하면 오른쪽에 Node(Layer)에 대한 세부 설정 값들을 확인할 수 있습니다.

몇몇 프레임워크 별로 sample을 이렇게 제공해줍니다!

open을 누르면 browser를 이용하여 볼 수 있네요!

P.S.

  • 피곤…
donaricano-btn
도움이 되셨다면 몰랑이에게 밀크티를...!
더 다양한 포스팅을 채우도록 노력할게요!
comments powered by Disqus