!13749 [MS][LITE]append some models to the entrance guard

From: @probiotics_53
Reviewed-by: 
Signed-off-by:
This commit is contained in:
mindspore-ci-bot 2021-03-30 19:06:00 +08:00 committed by Gitee
commit 37095afa4b
11 changed files with 46 additions and 2 deletions

View File

@ -103,6 +103,9 @@ int LiteKernel::PreProcess() {
for (auto *output : this->out_tensors()) {
MS_ASSERT(output != nullptr);
if (desc_.data_type == kNumberTypeFloat16 && output->data_type() == kNumberTypeFloat32) {
output->set_data_type(kNumberTypeFloat16);
}
if (output->ElementsNum() >= lite::MAX_MALLOC_SIZE / static_cast<int>(sizeof(int64_t))) {
MS_LOG(ERROR) << "The size of output tensor is too big";
return RET_ERROR;

View File

@ -80,6 +80,7 @@ hdc_emotion
hdc_fivembnet
hdc_isface
hdc_mobilenetface
hdc_retinaface
hdc_resnet
ml_video_edit_detect
ml_video_edit_hairSeg_have_imageProcessLayer_interpTo145_20210121
@ -87,3 +88,4 @@ ml_video_edit_have_imageProcessLayer_interpTo145_20201015
ml_video_edit_MnetN367_extract_1010_pay
ml_video_edit_reid
ml_video_edit_v10_best_model_nomean_20200723
ml_handpose

View File

@ -74,7 +74,9 @@ hdc_emotion 0.5
hdc_fivembnet 0.5
hdc_isface 0.5
hdc_mobilenetface 7.5
hdc_retinaface 14
hdc_resnet 7
hdc_Face_Aesthetic_MTI_Aesthetic 22
ml_video_edit_detect 2.5
ml_video_edit_hairSeg_have_imageProcessLayer_interpTo145_20210121 0.5
ml_video_edit_have_imageProcessLayer_interpTo145_20201015 0.5
@ -82,3 +84,5 @@ ml_video_edit_MnetN367_extract_1010_pay 1
ml_video_edit_person_divison_pic 0.5
ml_video_edit_reid 1
ml_video_edit_v10_best_model_nomean_20200723 5
ml_video_edit_img_segment 3
ml_video_edit_video_segment_gauss_adaptis_part1 5

View File

@ -27,3 +27,4 @@ quant_aware_bank_card_detection_inception.onnx
quant_aware_bank_card_recognition_fcny.onnx
quant_aware_identify_card_detect.onnx
tiny-yolov3-11.onnx;2;1,416,416,3:1,2
ml_video_edit_art_transfer.onnx;3

View File

@ -62,5 +62,17 @@ ml_video_edit_have_imageProcessLayer_interpTo145_20201015 0.5
ml_video_edit_MnetN367_extract_1010_pay 0.5
ml_video_edit_reid 0.5
ml_video_edit_v10_best_model_nomean_20200723 8
ml_edu_kit_hand_key_position.onnx 2
ml_video_edit_art_generate.onnx 0.5
#hdc_ocr_attention.onnx 0.5 #too many subgraphs
#hdc_ocr_detect.onnx 30 #too many subgraphs
ml_edu_kit_hand_detection.onnx 1
ml_edu_kit_hand_key_position.onnx 2
#ml_video_edit_oneclick_adaptis.pb #too many subgraphs
ml_video_edit_enhance.pb 1.5
ml_video_edit_video_segment_gauss_adaptis_part2_pb2tflite.tflite 10 2
ml_video_edit_img_segment_adaptise_pb2tflite.tflite 0.5 2
#hdc_retinaface #too many subgraphs
# hdc_age_medium need further analysis, now add for guarding process
hdc_age_medium 504
gts_detect_5k_tf115.tflite 319
ml_video_edit_art_transfer.onnx 3 3

View File

@ -53,5 +53,6 @@ hdc_resnet_1w_class.onnx
ml_video_edit_imitate_filter.onnx
hdc_ocr_attention.onnx
hdc_ocr_detect.onnx
ml_edu_kit_hand_detection.onnx
ml_edu_kit_hand_key_position.onnx
ml_video_edit_art_generate.onnx

View File

@ -38,4 +38,18 @@ residual_distill_res34_cifar10_bs_1_update.onnx 2
residual_distill_res50_cifar10_bs_1_update.onnx 2
hdc_ocr_attention.onnx 1
hdc_ocr_detect.onnx 30 #one of the output has small values
hdc_Face_Emotion_MTI_Aesthetic.onnx 145
ml_edu_kit_hand_detection.onnx 2
ml_edu_kit_hand_key_position.onnx 2
ml_video_edit_judge.onnx 12
ml_video_edit_vignet.onnx 1.5
ml_video_edit_imitate_filter.onnx 105
#ml_video_edit_art_generate.onnx 0.5 # not suitable for fp16
ml_video_edit_style_transfer_autoportrait.onnx 2
ml_video_edit_style_transfer_candy.onnx 2
ml_video_edit_style_transfer_gongnongbing.onnx 2
ml_video_edit_style_transfer_starry.onnx 2
hdc_Face_Landmark5_MTI_Aesthetic.onnx 0.5
hdc_Image_Aesthetic_MTI_Aesthetic.onnx 0.5
hdc_resnet_1w_class.onnx 6
hdc_mobilenet_1w_class.onnx 20

View File

@ -1,2 +1,4 @@
decoder_step_201217_modified.pb 5
encoder_0111_control_flow.pb 4;1:1,44:1:1%9
ml_video_edit_generate_filter.pb 1;%2
ml_video_edit_enhance.pb 1;%2

View File

@ -1,7 +1,7 @@
lite-model_arbitrary-image-stylization-inceptionv3_fp16_transfer_1.tflite;2
magenta_arbitrary-image-stylization-v1-256_fp16_transfer_1.tflite;2
# albert_lite_base_squadv1_1.tflite;3
# mobilebert_1_default_1.tflite;3
mobilebert_1_default_1.tflite;3
porseg_tmp.onnx;2
ml_video_edit_img_segment_adaptise.pb;2
ml_video_edit_img_segment_adaptise_pb2tflite.tflite;2

View File

@ -2047,6 +2047,10 @@ function Run_npu() {
# Run npu converted models:
while read line; do
model_line_info=${line}
if [[ $model_line_info == \#* ]]; then
continue
fi
model_name=`echo ${line}|awk -F ' ' '{print $1}'`
accuracy_limit=`echo ${line}|awk -F ' ' '{print $2}'`
input_num=`echo ${line}|awk -F ' ' '{print $3}'`

View File

@ -175,6 +175,7 @@ generateOpsList
getCommonFile
# get src/ops
getOpsFile "Registry\(schema::PrimitiveType_" "${MINDSPORE_HOME}/mindspore/lite/src/ops" "prototype" &
getOpsFile "REG_POPULATE\(PrimitiveType_" "${MINDSPORE_HOME}/mindspore/lite/src/ops" "prototype" &
getOpsFile "REG_KERNEL\(.*?, kNumberTypeFloat32, PrimitiveType_" "${MINDSPORE_HOME}/mindspore/lite/src/runtime/kernel/arm" "kNumberTypeFloat32" &
getOpsFile "REG_KERNEL\(.*?, kNumberTypeFloat16, PrimitiveType_" "${MINDSPORE_HOME}/mindspore/lite/src/runtime/kernel/arm" "kNumberTypeFloat16" &
getOpsFile "REG_KERNEL\(.*?, kNumberTypeInt8, PrimitiveType_" "${MINDSPORE_HOME}/mindspore/lite/src/runtime/kernel/arm" "kNumberTypeInt8" &