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

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

From: @sishuikang
Reviewed-by: @zhanghaibo5,@hangangqiang
Signed-off-by: @zhanghaibo5
This commit is contained in:
mindspore-ci-bot 2021-02-26 14:57:52 +08:00 committed by Gitee
parent 6d2a34216f 702833b73c
commit e104b6b23f
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
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();