.. caffeModelSummarizer. caffeModelSummarizer ==================== 概要 ---- caffeModelSummarizerは、DNN開発プラットフォームcaffeがネットワークの定義ファイルとして読み込むprototxtを読み込んで、 ネットワーク構造をサマライズするためのユーティリティです。 CSV形式のExcel等で読み込めるサマリファイル以外に、その情報をGraphvizで描画可能なダイアグラムとしても出力することができます。 DNNネットワークを画像化できる他ツールも存在しますが、本ツールの特徴は各レイヤの係数量、ops値なども表示されることです。 リビジョン ---------- caffeModelSummarizer.py v01r02 コマンドライン -------------- .. code-block:: none python caffeModelSummarizer.py [option [option_parameter]]... 以下のようなオプションをつけて動作させることで所望の結果を得ることができます。 **コマンドラインオプション** +-----------+------------------------------------+ |オプション | 意味 | +===========+====================================+ |-h | ヘルプ表示 | +-----------+------------------------------------+ |--PROT pt | 入力であるprototxtファイル名を指定 | +-----------+------------------------------------+ |--CSV csv | レイヤ毎の集計出力ファイル名 | +-----------+------------------------------------+ |--DOT d | Graphviz用描画スクリプト | +-----------+------------------------------------+ |--ICH i | 入力チャネルサイズ指定 | +-----------+------------------------------------+ |--IH h | 入力縦サイズ | +-----------+------------------------------------+ |--IW w | 入力横サイズ | +-----------+------------------------------------+ |--NAME n | Graph Name指定(prototxt内不在の時) | +-----------+------------------------------------+ |-v | 処理の途中経過を詳細に出力 | +-----------+------------------------------------+ |-V | バージョンのみ表示して終了 | +-----------+------------------------------------+ |-d | デバッグ用オプション | +-----------+------------------------------------+ 構造 ---- .. automodule:: caffeModelSummarizer .. autoclass:: caffeModelSummarizer.caffeProtoTxtReader :members: .. autoclass:: caffeModelSummarizer.csvFileWriter :members: .. autoclass:: caffeModelSummarizer.DOTwriterDNNdataFlow :members: