ネットワークの変換 - 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
を開いてドキュメントをご覧ください。