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

From: @wang_shaocong
Reviewed-by: 
Signed-off-by:
pull/11644/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit 8784aa3b2f

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

@ -34,3 +34,10 @@ ml_video_edit_style_transfer_starry.onnx 11
porseg_tmp.onnx 1 2 porseg_tmp.onnx 1 2
ml_video_edit_Mnet 1.5 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

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

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

@ -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.pb;2
ml_video_edit_video_segment_gauss_adaptis_part2_pb2tflite.tflite;2 ml_video_edit_video_segment_gauss_adaptis_part2_pb2tflite.tflite;2
tiny-yolov3-11.onnx;2;1,416,416,3:1,2 tiny-yolov3-11.onnx;2;1,416,416,3:1,2
ml_video_edit_person_divison_pic;2

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

Loading…
Cancel
Save