Versions Compared

Key

  • This line was added.
  • This line was removed.
  • Formatting was changed.

...

Table of Contents
minLevel1
maxLevel4

環境構築

...

環境構築手順はこちらをご参照ください。

モデルを変換する

モデル変換を行います。

引数の「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

...