Netron - 機械学習のネットワークを可視化
機械学習ではフォーマットの標準化が進んでいます。これまでTensorflowやCaffe、Chainerなど様々なソフトウェアがありましたが、生成されるモデルがONNXというフォーマットに統一されつつあります(他にもあるようですが)。 そんなONNXフォーマットのモデルをビジュアル化できるソフトウェアがNetronです。
Netronの使い方
ドロップしたところです。これは4か9を判別するモデルです。
拡大もできます。
パラメータも閲覧できます。
Netronを使うことでPythonのコードではなく、ビジュアル化されたモデルを通じて機械学習のネットワークを学ぶことができます。各ノードにおける出力の数も出ており、ネットワークの詳細が分かりやすくなるでしょう。
NetronはJavaScript製のオープンソース・ソフトウェア(MIT License)です。
Netron lutzroeder/netron: Visualizer for deep learning and machine learning models