Browse Source

add tflite and caffe models

tags/v0.7.0-beta
meng_chunyang 5 years ago
parent
commit
631aa8cf46
3 changed files with 6 additions and 3 deletions
  1. +2
    -2
      mindspore/lite/test/models_caffe.cfg
  2. +3
    -0
      mindspore/lite/test/models_tflite.cfg
  3. +1
    -1
      mindspore/lite/tools/converter/parser/tflite/tflite_activation_parser.cc

+ 2
- 2
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
- 0
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


+ 1
- 1
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) {


Loading…
Cancel
Save