Fix: throw error on NCNN extraction fail

jolibrain · Dec 18, 2018 · 569800a · 569800a
1 parent 6b34a44
commit 569800a
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/backends/ncnn/ncnnlib.cc b/src/backends/ncnn/ncnnlib.cc
@@ -124,10 +124,14 @@ namespace dd
             bbox = true;
 
         // Extract detection or classification
+        int ret = 0;
         if (bbox == true) {
-            ex.extract("detection_out", inputc._out);
+            ret = ex.extract("detection_out", inputc._out);
         } else {
-            ex.extract("prob", inputc._out);
+            ret = ex.extract("prob", inputc._out);
+        }
+        if (ret == -1) {
+            throw MLLibInternalException("NCNN internal error");
         }
 
         std::vector<APIData> vrad;