Browse Source

!12580 [MS][Lite]fix mindspore to MindSpore on image classification

From: @sishuikang
Reviewed-by: @zhanghaibo5,@hangangqiang
Signed-off-by: @zhanghaibo5
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 4 years ago
parent
commit
e104b6b23f
1 changed files with 4 additions and 4 deletions
  1. +4
    -4
      model_zoo/official/lite/image_classification/app/src/main/cpp/MindSporeNetnative.cpp

+ 4
- 4
model_zoo/official/lite/image_classification/app/src/main/cpp/MindSporeNetnative.cpp View File

@@ -215,7 +215,7 @@ char *CreateLocalModelBuffer(JNIEnv *env, jobject modelBuffer) {
} }


/** /**
* To process the result of mindspore inference.
* To process the result of MindSpore inference.
* @param msOutputs * @param msOutputs
* @return * @return
*/ */
@@ -363,7 +363,7 @@ Java_com_mindspore_classification_gallery_classify_TrackingMobile_loadModel(JNIE
return (jlong) nullptr; return (jlong) nullptr;
} }


// To create a mindspore network inference environment.
// To create a MindSpore network inference environment.
void **labelEnv = new void *; void **labelEnv = new void *;
MSNetWork *labelNet = new MSNetWork; MSNetWork *labelNet = new MSNetWork;
*labelEnv = labelNet; *labelEnv = labelNet;
@@ -415,7 +415,7 @@ Java_com_mindspore_classification_gallery_classify_TrackingMobile_runNet(JNIEnv
inputDims.width = lite_norm_mat_cut.width_; inputDims.width = lite_norm_mat_cut.width_;
inputDims.height = lite_norm_mat_cut.height_; inputDims.height = lite_norm_mat_cut.height_;


// Get the mindsore inference environment which created in loadModel().
// Get the MindSpore inference environment which created in loadModel().
void **labelEnv = reinterpret_cast<void **>(netEnv); void **labelEnv = reinterpret_cast<void **>(netEnv);
if (labelEnv == nullptr) { if (labelEnv == nullptr) {
MS_PRINT("MindSpore error, labelEnv is a nullptr."); MS_PRINT("MindSpore error, labelEnv is a nullptr.");
@@ -451,7 +451,7 @@ Java_com_mindspore_classification_gallery_classify_TrackingMobile_runNet(JNIEnv
} }


/** /**
* Get the mindspore inference results.
* Get the MindSpore inference results.
* Return the map of output node name and MindSpore Lite MSTensor. * Return the map of output node name and MindSpore Lite MSTensor.
*/ */
auto names = mSession->GetOutputTensorNames(); auto names = mSession->GetOutputTensorNames();


Loading…
Cancel
Save