ディープラーニング

MMdnnを試す

モデル変換機能

MMdnnは様々な機械学習フレームワークで作成したニューラルネットワークモデルを相互変換できるツールです。

 

出展:Github: https://github.com/Microsoft/MMdnn

 

現在ImageNetのモデルで変換テスト済みのリストも公開されています。

MMdnn_ImageNet
RNNモデル進行中なので今後も楽しみです。

モデルネットワークの可視化

そして何より面白いなと思ったのが学習中の状況を可視化できるツールはいろいろありますが

このMMdnnはネットワーク自体を可視化できるのが面白い!

MMdnn model visualizer

図:kerasのinception v3モデル

 

MMdnnのインストール

ではさっそくインストールしましょう。

とはいってもインストールするにはたったの1行

pip install mmdnn

 

主に使うコマンドとしては

mmdownload(モデルのダウンロード)

mmconvert(モデルの変換)

mmtoir(ネットワークデータの出力)

です。

 

各コマンドの引数はマニュアルを〜と探してみたのですが

それぞれのソースコードに書いてあるのでそれを見るしかなさそうです。

例:mmconvert

https://github.com/Microsoft/MMdnn/blob/master/mmdnn/conversion/_script/convert.py