Netron - 機械学習のネットワークを可視化

機械学習ではフォーマットの標準化が進んでいます。これまでTensorflowやCaffe、Chainerなど様々なソフトウェアがありましたが、生成されるモデルがONNXというフォーマットに統一されつつあります（他にもあるようですが）。

そんなONNXフォーマットのモデルをビジュアル化できるソフトウェアがNetronです。

Netronの使い方

ドロップしたところです。これは4か9を判別するモデルです。

拡大もできます。

パラメータも閲覧できます。

Netronを使うことでPythonのコードではなく、ビジュアル化されたモデルを通じて機械学習のネットワークを学ぶことができます。各ノードにおける出力の数も出ており、ネットワークの詳細が分かりやすくなるでしょう。

NetronはJavaScript製のオープンソース・ソフトウェア（MIT License）です。