fix result blob in squeezenet example

tcye · Mar 4, 2018 · ecaadb2 · ecaadb2
1 parent 68f0169
commit ecaadb2
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 9 deletions.
diff --git a/examples/squeezencnn/jni/Android.mk b/examples/squeezencnn/jni/Android.mk
@@ -1,7 +1,7 @@
 LOCAL_PATH := $(call my-dir)
 
 # change this folder path to yours
-NCNN_INSTALL_PATH := /home/nihui/dev/qqfacecnn/ncnn/build-android-armv7/install
+NCNN_INSTALL_PATH := /home/nihui/dev/ncnn/build-android-armv7/install
 
 include $(CLEAR_VARS)
 LOCAL_MODULE := ncnn

diff --git a/examples/squeezencnn/jni/squeezencnn_jni.cpp b/examples/squeezencnn/jni/squeezencnn_jni.cpp
@@ -143,11 +143,10 @@ JNIEXPORT jstring JNICALL Java_com_tencent_squeezencnn_SqueezeNcnn_Detect(JNIEnv
         ncnn::Mat out;
         ex.extract(squeezenet_v1_1_param_id::BLOB_prob, out);
 
-        cls_scores.resize(out.c);
-        for (int j=0; j<out.c; j++)
+        cls_scores.resize(out.w);
+        for (int j=0; j<out.w; j++)
         {
-            const float* prob = out.channel(j);
-            cls_scores[j] = prob[0];
+            cls_scores[j] = out[j];
         }
     }
 

diff --git a/examples/squeezenet.cpp b/examples/squeezenet.cpp
@@ -39,11 +39,10 @@ static int detect_squeezenet(const cv::Mat& bgr, std::vector<float>& cls_scores)
     ncnn::Mat out;
     ex.extract("prob", out);
 
-    cls_scores.resize(out.c);
-    for (int j=0; j<out.c; j++)
+    cls_scores.resize(out.w);
+    for (int j=0; j<out.w; j++)
     {
-        const float* prob = out.channel(j);
-        cls_scores[j] = prob[0];
+        cls_scores[j] = out[j];
     }
 
     return 0;