update ascend cloud gate model

This commit is contained in:
wangwenzhe 2022-11-21 09:31:51 +08:00 committed by wangwenzhe_oo
parent 438c18f5f4
commit f30f0b11d5
3 changed files with 15 additions and 2 deletions

View File

@ -7,6 +7,9 @@ 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
EI_yolox.onnx;1:input;1,3,640,640;;offline_resize 5
EI_screenshot_text_location_220613_100.onnx;1:input_images;1,3,1024,1024;;offline_resize 5
EI_resnet18_batch_size1.onnx;1:input_node;1,6,224,224;;offline_resize 5
#mindir
mindspore_uniir_mobilenetv2.mindir 0.5
@ -27,3 +30,13 @@ 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
# cbg_ai_ocr_language_classify_latin.pb;1:data;2,48,1,50;;offline_resize 5
cbg_ai_ocr_recognize_chinese_english.pb;1:input_0;1,2048,2048,1;;offline_resize 5
cbg_ai_ocr_recognize_chinese_english_vertical.pb;1:input_0;1,2048,2048,1;;offline_resize 5
cbg_ai_ocr_recognize_japanese_korean.pb;1:input_0;1,2048,2048,1;;offline_resize 5
cbg_ai_text_filing_inpainting.pb;2:input_images,input_masks;1,32,32,3:1,32,32,1;;offline_resize 10
open_source_inception_resnet_v2.pb;1:input;2,299,299,3;;offline_resize 5
open_source_mobilenet_v1_10_224_frozen.pb;1:input;2,224,224,3;;offline_resize 5
# EI_screenshot_direction.pb;1:input_1;1,64,512,3;;offline_resize 10
EI_screenshot_angle.pb;1:input_images;1,1024,1024,3;;offline_resize 5
# EI_screenshot_ocr_220613_batch32_textlen24.pb;1:img_data;1,32,256,3;;offline_resize 5

View File

@ -70,7 +70,7 @@ function Run_Benchmark() {
fi
input_files=""
output_file=""
data_path=${model_data_path}'/data/'
data_path=${model_data_path}'/models/hiai/input_output/'
if [[ ${model_type} == "mindir" || ${model_type} == "ms" ]]; then
if [[ ${input_num} == "" || ${input_num} == 1 ]]; then
input_files=${data_path}'input/'${model_name}'.ms.bin'

View File

@ -41,7 +41,7 @@ rm -rf ${ms_models_path}
mkdir -p ${ms_models_path}
model_data_path=/home/workspace/mindspore_dataset/mslite
models_path=${model_data_path}/models
models_path=${model_data_path}/models/hiai
if [[ ${backend} =~ "lite" ]]; then
compile_type="lite"