Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add tflite and caffe models

This commit is contained in:
meng_chunyang 2020-08-21 20:37:01 +08:00
parent 2b923e5674
commit 631aa8cf46
3 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/lite/test/models_caffe.cfg
View File

@ -17,10 +17,10 @@ mtk_isface
mtk_landmark
mtk_pose_tuku
mtk_face_recognition_v1
#mtk_2012_ATLANTA_10class_20190614_v4.1
mtk_2012_ATLANTA_10class_20190614_v41
mtk_detect-deeper-halfdeeper-mbv1-lastearlySSD-shortcut-400-400_nopostprocess_simplified
detect-deeper-halfdeeper-mbv1-shortcut-400-400_nopostprocess_simplified
#hiai_face_detect_rfb
hiai_face_detect_rfb
hiai_face_isface
hiai_face_landmark
hiai_face_pose_tuku

3
mindspore/lite/test/models_tflite.cfg
View File

@ -64,6 +64,9 @@ mtk_model_normalize_object_scene_ps_20200519_f16.tflite
mtk_age_gender_fp16.tflite
mtk_model_face_dress_fp16.tflite
mtk_AADB_HADB_MBV2_model_f16.tflite
mtk_AADB_HADB_MBV3_model_f16.tflite
mtk_model_emotions_0725_fp16.tflite
magenta_arbitrary-image-stylization-v1-256_fp16_prediction_1.tflite
ml_object_detect.tflite
hiai_cpu_face_emotion.tflite
hiai_cpu_face_gazing.tflite

2
mindspore/lite/tools/converter/parser/tflite/tflite_activation_parser.cc
View File

@ -63,7 +63,7 @@ STATUS TfliteActivationParser::Parse(const std::unique_ptr<tflite::OperatorT> &t
attr->type = schema::ActivationType_SIGMOID;
} else if (std::strcmp(node_name, "HardSwish") == 0) {
MS_LOG(DEBUG) << "parse TfliteHardSwishParser";
attr->type = schema::ActivationType_SIGMOID;
attr->type = schema::ActivationType_HSWISH;
} else if (std::strcmp(node_name, "LeakyRelu") == 0) {
const auto &tflite_attr = tflite_op->builtin_options.AsLeakyReluOptions();
if (tflite_attr == nullptr) {