!33788 ascend smoke test

Merge pull request !33788 from zhengyuanhua/master
This commit is contained in:
i-robot 2022-05-05 02:07:06 +00:00 committed by Gitee
commit dbfa01dfea
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
2 changed files with 40 additions and 4 deletions

View File

@ -2,7 +2,28 @@
# model_name;input_info;input_shapes;mode accuracy_limit/CONVERTER
# Run in Ascend
#onnx
hdc_resnet_1w_class.onnx 3
googlenet-9.onnx;1:data_0;1,3,224,224;;offline_resize 5
cbg_ai_gender_resnet34_mutiscal_v1_1012.onnx;1:input.1;1,1,300,64;;offline_resize 5
hdc_efficientnet_b3_1w_class.onnx;1:input.1;1,3,224,224;;offline_resize 5
#mindir
mindspore_uniir_mobilenetv2.mindir 0.5
#caffe
hdc_resnet 4
#cbg_ai_ocr_direction.pb;1:input;1,2048,2048,1;;offline_resize 2
machine_vision_mobile_net101_resnet 5
ml_video_edit_img_segment 5
mtk_face_recognition_v1 5
#tf
cbg_ai_ocr_direction.pb;1:input;1,32,32,1;;offline_resize 2
cbg_ai_ocr_detect_straight.pb;1:input;1,32,32,3;;offline_resize 5
cbg_ai_ocr_inception_curve.pb;1:input;1,960,960,3;;offline_resize 5
cbg_ai_ocr_language_classify.pb;1:input_0;2,32,512,1;;offline_resize 5
cbg_ai_ocr_recognize_latin.pb;1:input_0;1,64,64,1;;offline_resize 5
open_source_inception_v3.pb;1:input;2,299,299,3;;offline_resize 5
open_source_mobilenet_v2.pb;1:Placeholder;1,224,224,3;;offline_resize 5
open_source_squeeze_net.pb;1:Placeholder;2,224,224,3;;offline_resize 5
open_source_densenet.pb;1:Placeholder;2,224,224,3;;offline_resize 5

View File

@ -58,19 +58,34 @@ function Run_Benchmark() {
fi
echo "Benchmarking ${model_name} ......"
model_type=${model_name##*.}
model_file=${ms_models_path}'/'${model_name}'.ms'
input_files=""
output_file=""
data_path=${model_data_path}'/data/'
if [[ ${input_num} == "" || ${input_num} == 1 ]]; then
if [[ ${model_type} == "mindir" ]]; then
if [[ ${input_num} == "" || ${input_num} == 1 ]]; then
input_files=${data_path}'input/'${model_name}'.ms.bin'
else
else
for i in $(seq 1 $input_num)
do
input_files=${input_files}${data_path}'input/'${model_name}'.ms.bin_'$i','
done
fi
output_file=${data_path}'output/'${model_name}'.ms.out'
else
if [[ ${input_num} == "" || ${input_num} == 1 ]]; then
input_files=${data_path}'input/'${model_name}'.bin'
else
for i in $(seq 1 $input_num)
do
input_files=${input_files}${data_path}'input/'${model_name}'.bin_'$i','
done
fi
output_file=${data_path}'output/'${model_name}'.out'
fi
output_file=${data_path}'output/'${model_name}'.ms.out'
# set accuracy limitation
acc_limit="0.5"