p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

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

From: @wang_shaocong
Reviewed-by: 
Signed-off-by:
This commit is contained in:
mindspore-ci-bot 2021-02-01 10:32:44 +08:00 committed by Gitee
parent 27e895ea82 85f76a3215
commit 8784aa3b2f
6 changed files with 27 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
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

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

20
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