機械学習モデル可視化ツール「Netron」を使ってみる

| 2 min read
Author: shuichi-takatsu shuichi-takatsuの画像

前回、ディープラーニングモデルのオープンフォーマット「ONNX(Open Neural Network Exchange)」を紹介しました。
その時に機械学習モデル可視化ツール「Netron」の存在を知りました。
今回は「Netron」を紹介したいと思います。

Netron って何?

#

Netronは、機械学習モデルを可視化するツールです。
ブラウザを含むクロスプラットフォームで動作します。

2023/02/05現在、Netron は、ONNX、TensorFlow Lite、Caffe、Keras、Darknet、PaddlePaddle、ncnn、MNN、Core ML、RKNN、MXNet、MindSpore Lite、TNN、Barracuda、Tengine、CNTK、TensorFlow.js、Caffe2、UFF など非常に多くの機械学習モデルをサポートしています。

Netron のインストール

#

自分の環境にあったインストールモジュールをここからダウンロードします。

最新版は V6.5.3 のようです。
筆者の環境はWindows10なので、Windows用のインストーラーをダウンロードします。

ダウンロードしたEXEファイルを実行すると、すぐにインストールが開始されます。

インストールが終了すると「Netron」が起動してきました。

機械学習モデルの可視化

#

アプリケーションの左上の「File」-「Open」を選択し、前回に使用した「VGG19」のONNXモデルを選択してみます。

VGG19機械学習モデルの構造が表示されました。

アプリケーションの左上の「View」を押して、機能リストを表示させます。
その中から「Show Attributes」を選択します。

モデル中の各レイヤの詳細が表示されました。

レイヤの一つをマウスでクリックしてみます。
クリックしたレイヤのプロパティが右側のペインに表示されました。

ブラウザ版を使う

#

Netron はブラウザでも動作するので、次のURLからブラウザ版Netronを開きます。
ブラウザに次のような画面が表示されます。

「Open Model…」をクリックし、先程と同様に「VGG19」のONNXモデルを選択してみます。

VGG19機械学習モデルの構造が表示されました。

画面左上の「ハンバーガーボタン」を押すと、機能リストが表示されます。

リストの一番上の「Properties…」を選択します。
表示したモデルのプロパティが表示されました。

まとめ

#

Netron はビューアなので機械学習モデルを修正・変更することは出来ませんが、自分で作成した機械学習モデルやインターネットからダウンロードした学習済み機械学習モデルの構造を簡単に可視化できるので、ディープラーニングを学習するための強力なツールになると思います。

豆蔵デベロッパーサイト - 先週のアクセスランキング
  1. 基本から理解するJWTとJWT認証の仕組み (2022-12-08)
  2. AWS認定資格を12個すべて取得したので勉強したことなどをまとめます (2022-12-12)
  3. Nuxt3入門(第4回) - Nuxtのルーティングを理解する (2022-10-09)
  4. Backstageで開発者ポータルサイトを構築する - 導入編 (2022-04-29)
  5. Nuxt3入門(第8回) - Nuxt3のuseStateでコンポーネント間で状態を共有する (2022-10-28)
  6. Viteベースの高速テスティングフレームワークVitestを使ってみる (2022-12-28)
  7. ORマッパーのTypeORMをTypeScriptで使う (2022-07-27)
  8. Nuxt3入門(第1回) - Nuxtがサポートするレンダリングモードを理解する (2022-09-25)
  9. GitHub Actions - 構成変数(環境変数)が外部設定できるようになったので用途を整理する (2023-01-16)
  10. Jest再入門 - 関数・モジュールモック編 (2022-07-03)