forked from mindspore-Ecosystem/mindspore
!33788 ascend smoke test
Merge pull request !33788 from zhengyuanhua/master
This commit is contained in:
commit
dbfa01dfea
|
@ -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
|
||||
|
|
|
@ -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"
|
||||
|
|
Loading…
Reference in New Issue