Browse Source

!11644 [MSLITE] Add ci test for several models.

From: @wang_shaocong
Reviewed-by: 
Signed-off-by:
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 4 years ago
parent
commit
8784aa3b2f
6 changed files with 27 additions and 10 deletions
  1. +2
    -2
      mindspore/lite/test/models_caffe.cfg
  2. +8
    -1
      mindspore/lite/test/models_npu.cfg
  3. +2
    -0
      mindspore/lite/test/models_onnx.cfg
  4. +1
    -0
      mindspore/lite/test/models_tf.cfg
  5. +1
    -0
      mindspore/lite/test/models_with_several_inputs_or_without_outputs.cfg
  6. +13
    -7
      mindspore/lite/test/run_benchmark_nets.sh

+ 2
- 2
mindspore/lite/test/models_caffe.cfg View File

@@ -68,8 +68,8 @@ ml_location_scene_division
ml_tabel_recog
ml_text_division
6c_seg_nomean_20200610
ml_video_edit_person_divison
ml_video_edit_img_segment
ml_video_edit_video_segment_gauss_adaptis_part1
ml_video_edit_Mnet
ml_video_edit_hairSeg_have_imageProcessLayer_interpTo145
ml_video_edit_hairSeg_have_imageProcessLayer_interpTo145
ml_video_edit_person_divison_video

+ 8
- 1
mindspore/lite/test/models_npu.cfg View File

@@ -33,4 +33,11 @@ ml_video_edit_style_transfer_gongnongbing.onnx 10
ml_video_edit_style_transfer_starry.onnx 11
porseg_tmp.onnx 1 2
ml_video_edit_Mnet 1.5
ml_video_edit_hairSeg_have_imageProcessLayer_interpTo145 0.5
ml_video_edit_hairSeg_have_imageProcessLayer_interpTo145 0.5
ml_video_edit_img_segment 1
ml_video_edit_video_segment_gauss_adaptis_part1 2
ml_video_edit_generate_filter.pb 1
ml_video_edit_img_segment_adaptise.pb 0.5 2
ml_video_edit_video_segment_gauss_adaptis_part2.pb 3 2
ml_video_edit_person_divison_pic 8 2
ml_video_edit_person_divison_video 0.5

+ 2
- 0
mindspore/lite/test/models_onnx.cfg View File

@@ -42,3 +42,5 @@ ml_video_edit_style_transfer_autoportrait.onnx
ml_video_edit_style_transfer_candy.onnx
ml_video_edit_style_transfer_gongnongbing.onnx
ml_video_edit_style_transfer_starry.onnx
ml_video_edit_judge.onnx
ml_video_edit_vignet.onnx

+ 1
- 0
mindspore/lite/test/models_tf.cfg View File

@@ -5,3 +5,4 @@ encoder_201228.pb 3;1:1,22:1
female_model_step2_int16_noiseout.pb 66
encoder_0111_control_flow.pb 4;1:1,44:1:1
encoder_0111.pb 4;1:1,44:1:1
ml_video_edit_generate_filter.pb 1

+ 1
- 0
mindspore/lite/test/models_with_several_inputs_or_without_outputs.cfg View File

@@ -36,3 +36,4 @@ ml_video_edit_img_segment_adaptise_pb2tflite.tflite;2
ml_video_edit_video_segment_gauss_adaptis_part2.pb;2
ml_video_edit_video_segment_gauss_adaptis_part2_pb2tflite.tflite;2
tiny-yolov3-11.onnx;2;1,416,416,3:1,2
ml_video_edit_person_divison_pic;2

+ 13
- 7
mindspore/lite/test/run_benchmark_nets.sh View File

@@ -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


Loading…
Cancel
Save