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

スクリプトを実行する

ネットワーク変換ツールはPython で書かれています。変換ツールを使うには、単に下記のコマンドをコンソールに打ち込んで下さい。

$ cd dv-sdk/tool
$ python3 convertor.py ../application/model/caffe_mobilenet.ini

設定を入力する

ネットワーク変換ツールは.ini ファイルで設定をします。

[INPUT]
name = CaffeMobileNet                   ;<- Put your network name here
definition = mobilenet_deploy.prototxt  ;<- Network definition
data = mobilenet.caffemodel             ;<- Network weights
origin = CAFFE                          ;<- Network origin type

[OUTPUT]
output_folder = "../"            ;<- Output folder root. Actual output is in output_folder/name
generate_source = 1              ;If enabled, generate the .cpp and .h source files for the network
generate_doxy 	= 1              ;If enabled, generate the doxygen documentation for the generate code 
generate_dot 	= 1              ;If enabled, add the resulting network graph to documentation using dot format
...

設定可能な全項目についてはネットワーク変換ツールへの入力ファイル をご覧ください。

生成されるファイル

  • CaffeMobileNet_gen.cpp(.h): 変換されたネットワークの設定をもつソースコードです。
  • CaffeMobileNet_weights.bin: パックされた重みのファイルです。
  • doc folder: Doxygen ドキュメントのソースです。

Doxygen とGraphiviz がインストールされていれば、以下のコマンドで変換されたネットワークに関するドキュメントを生成できます。

$ cd dv-sdk/application/CaffeMobileNet/doc
$ doxygen

このコマンドの後でapplication/CaffeMobileNet/doc/html/index.html を開いてドキュメントをご覧ください。