tensorflowのモデルファイル( *.pb や *.tflite )を可視化する方法について。
tensorboard を使う方法などもあるが、最もお手軽と思われる netron を使用する。

[!NOTE] これやらなくても、https://lutzroeder.github.io/netron/ にアクセスすれば使える。

netron の インストール

netronをインストールする。
netron はtensorflowのバージョンに依存しない(そもそもtensorflow自体に依存してない)のでTensorflow1系、Tenorflow2系 どちらの環境にインストールしてもよい。

pip install netron

netron の 実行

netron を実行する。

netron --host 0.0.0.0 [--port xxxx]

--host オプションを指定しないとlocalhostからしか接続できないので、他のマシンから接続するときは指定する。
--port オプションのデフォルトは8080なので、変更したいときは指定する。

モデルファイルの表示

ブラウザ(Winマシンからで可)で実行したマシンのポート8008(またはオプションで指定したxxxx)に接続。
例：http://ncc-1701u.local:8080/

表示された画面でOpen Model…をクリックして表示するモデルファイルを選ぶ。
しばらく待つと表示される。
モデルファイルはブラウザからアップロードするので、ブラウザから参照できる場所になければならない(サーバ側にあってもダメ)。

*.pb ファイルが大きいとうまく表示できないみたい。