MRが楽しい

MRやVRについて学習したことを書き残す

NETRONを使って深層学習のモデルを可視化する

本日はNETRONの技術調査枠です。
NETRONを使って深層学習のモデルを可視化する方法についてです。

NETRON

NETRON(ネトロン)は、機械学習・深層学習のモデル(ニューラルネットワーク構造など)を可視化するためのツールです。
ONNX、TensorFlow Lite、Core ML、Keras、Caffe、Darknet、PyTorch、TensorFlow.js、Safetensors、NumPyなどをサポートしています。
github.com

インストール手順

以下のGitHubページのReleaseページを開きます。
github.com

ソフトをインストールする環境に合わせてインストーラを選択します。
今回、筆者はWindows環境にインストールするため、Netron-Setup-X.X.X.exeをダウンロードしました。
github.com

Netron-Setup-X.X.X.exeを実行するとインストールが実行され、NETRONが起動します。

アプリが起動すると以下の画面が表示されます。
モデルを開くには[Open Model..]をクリックします。

エクスプローラが開くのでモデルを選択します。

以下の通りモデルの構造が可視化して表示されます。