|
|
|
@ -293,20 +293,26 @@ function Run_Converter() {
|
|
|
|
|
tflite)
|
|
|
|
|
model_fmk="TFLITE"
|
|
|
|
|
;;
|
|
|
|
|
caffemodel)
|
|
|
|
|
model_name=${model_name%.*}
|
|
|
|
|
model_fmk="CAFFE"
|
|
|
|
|
;;
|
|
|
|
|
onnx)
|
|
|
|
|
model_fmk="ONNX"
|
|
|
|
|
;;
|
|
|
|
|
mindir)
|
|
|
|
|
model_fmk="MINDIR"
|
|
|
|
|
;;
|
|
|
|
|
*)
|
|
|
|
|
model_type="caffe"
|
|
|
|
|
model_fmk="CAFFE"
|
|
|
|
|
;;
|
|
|
|
|
esac
|
|
|
|
|
echo ${model_name} >> "${run_converter_log_file}"
|
|
|
|
|
echo './converter_lite --fmk='${model_fmk}' --modelFile='${models_path}'/'${model_name}' --outputFile='${ms_models_path}'/'${model_name} >> "${run_converter_log_file}"
|
|
|
|
|
./converter_lite --fmk=${model_fmk} --modelFile=${models_path}/${model_name} --outputFile=${ms_models_path}/${model_name}
|
|
|
|
|
if [[ $model_fmk == "CAFFE" ]]; then
|
|
|
|
|
echo ${model_name} >> "${run_converter_log_file}"
|
|
|
|
|
echo './converter_lite --fmk='${model_fmk}' --modelFile='$models_path/${model_name}'.prototxt --weightFile='$models_path'/'${model_name}'.caffemodel --outputFile='${ms_models_path}'/'${model_name} >> "${run_converter_log_file}"
|
|
|
|
|
./converter_lite --fmk=${model_fmk} --modelFile=${models_path}/${model_name}.prototxt --weightFile=${models_path}/${model_name}.caffemodel --outputFile=${ms_models_path}/${model_name}
|
|
|
|
|
else
|
|
|
|
|
echo ${model_name} >> "${run_converter_log_file}"
|
|
|
|
|
echo './converter_lite --fmk='${model_fmk}' --modelFile='${models_path}'/'${model_name}' --outputFile='${ms_models_path}'/'${model_name} >> "${run_converter_log_file}"
|
|
|
|
|
./converter_lite --fmk=${model_fmk} --modelFile=${models_path}/${model_name} --outputFile=${ms_models_path}/${model_name}
|
|
|
|
|
fi
|
|
|
|
|
if [ $? = 0 ]; then
|
|
|
|
|
converter_result='converter '${model_type}' '${model_name}' pass';echo ${converter_result} >> ${run_converter_result_file}
|
|
|
|
|
else
|
|
|
|
|