...
Table of Contents | ||||
---|---|---|---|---|
|
環境構築
...
環境構築手順はこちらをご参照ください。
モデルを変換する
モデル変換を行います。
引数の「setting.conf」は変換するモデルに合わせてパラメータを変更してください。
Code Block |
---|
$ cd /home/cvtool/conversion/caffe
$ ./caffe_conversion.sh setting.conf |
変換後のモデルは以下のディレクトリに出力されます。
${OUTPUT_DIR}/${NET_NAME}/${PARSER_OPTION}/[モデル名]
Setting.conf仕様
...
Code Block |
---|
# Network Name NET_NAME=mobilenetv1 # Path to Deploy Prototxt DEPLOY_PROTOTXT=./sample/mobilenet_v1/mobilenet_deploy.prototxt # Path to Directory for (Deploy) Caffe Models MODEL_DIR=./sample/mobilenet_v1/models # Path to Directory for DRA Images DRA_IMAGE_DIR=../dra_img # Path to Directory for Output Data OUTPUT_DIR=./out # Quantization Mode # FIX8 : Fixed-point 8bit # FIX16 : Fixed-point 16bit # MIX : FIX8/FIX16 mixed PARSER_OPTION=FIX8 # Input Data Format (0:NHWC, 1:NCHW) IN_DATA_FORMAT=1 # Input Data Channel IN_DATA_CHANNEL=3 # Input Data Width IN_DATA_WIDTH=224 # Input Data Height IN_DATA_HEIGHT=224 # Input Data Mean Vector or Name of .binaryproto IN_MEAN=103.94,116.78,123.68 # Input Data Scale # IN_SCALE=1/Scale IN_SCALE=58.823529411 # RGB or BGR (0:RGB, 1:BGR) IS_BGR=1 # Input Layer Name IN_LAYER=data # Output Layers Name OUT_LAYER=mbox_loc,mbox_conf_flatten #cavalry version #if not specified -> "" CAVALRY_VER="2.1.7" # Unique preprocess # if use im2bin -> NONE # if use unique preprocess -> script path PREPRO=NONE PREPRO_ARG="" # Input file data format IN_DATA_FILEFORMAT=0,0,0,0 # Transpose indices(NONE:without transpose , 0,3,1,2:transpose (EX)) IN_DATA_TRANSPOSE=NONE |
...
deploy用のprototxtでinputレイヤが定義されていない場合は、以下のようにレイヤを追加してください。
...
モデルを変換する
...
モデル変換を行います。
引数の「setting.conf」は変換するモデルに合わせてパラメータを変更してください。
変換後のモデルは以下のディレクトリに出力されます。
${OUTPUT_DIR}/${NET_NAME}/${PARSER_OPTION}/[モデル名]
サンプルモデルの変換
...
ディレクトリを移動します。
Code Block |
---|
$ cd /home/cvtool/conversion/caffe |
サンプルモデルをダウンロードします。
Code Block |
---|
$ wget https://github.com/shicai/MobileNet-Caffe/blobraw/master/mobilenet.caffemodel $ mv mobilenet.caffemodel sample/mobilenet_v1/models |
...