!13749 [MS][LITE]append some models to the entrance guard
From: @probiotics_53 Reviewed-by: Signed-off-by:
This commit is contained in:
commit
37095afa4b
|
|
@ -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;
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
|
|
@ -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}'`
|
||||
|
|
|
|||
|
|
@ -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" &
|
||||
|
|
|
|||
Loading…
Reference in New Issue