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
* @return
*/
@@ -363,7 +363,7 @@ Java_com_mindspore_classification_gallery_classify_TrackingMobile_loadModel(JNIE
return (jlong) nullptr;
}

// To create a mindspore network inference environment.
// To create a MindSpore network inference environment.
void **labelEnv = new void *;
MSNetWork *labelNet = new MSNetWork;
*labelEnv = labelNet;
@@ -415,7 +415,7 @@ Java_com_mindspore_classification_gallery_classify_TrackingMobile_runNet(JNIEnv
inputDims.width = lite_norm_mat_cut.width_;
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);
if (labelEnv == 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.
*/
auto names = mSession->GetOutputTensorNames();


Loading…
Cancel
Save