ネットワーク変換ツール - DigitalMediaProfessionals/dv-sdk GitHub Wiki

ネットワーク変換ツールは、深層学習フレームワークよりエクスポートされたモデルや重みを、弊社のボードで実行できる形式に変換します。

セットアップ

環境設定 をみてPython 環境を用意してください。 AI FPGA モジュールではCPUのスピードやメモリ容量が非常に限られているので、変換はPC やサーバー上で行うことをお勧めします。

ネットワーク変換ツールの実行

ネットワーク変換ツールはPython で書かれています。変換ツールの実行には次のコマンドを実行してください。 設定ファイルの詳細については ツールの設定をご覧ください。

$ cd tool
$ python convertor.py <path to the input config .ini file>

ネットワーク変換ツールはネットワークを表すソースコードと重みファイル、Doxygen 関連のファイルを生成します。 詳細は変換されたネットワークの設定 をご覧ください。

ネットワーク変換ツールの基本的な手順は上の図の通りです。 変換ツールは入力されたネットワークを内部表現へと変換してから解析します。 変換ツールは複数の連続したノードを同じ層にまとめます。 現在の実装では解析はヒューリスティックであるため、結果が最適でないことがあります。

サポートしている深層学習フレームワーク

現在では、ネットワーク変換ツールはCaffe とKeras をサポートしています。 ただし入力サイズが未定のネットワークをそのまま変換することはできません。 サポートされているレイヤの種類についてはこちらをご覧ください。