Fix svace issue readModel NO_CATCH

Add try catch statement to handle exception in main.cpp and main_func.cpp

Signed-off-by: hyeonseok lee <hs89.lee@samsung.com>

diff --git a/Applications/TransferLearning/CIFAR_Classification/jni/main.cpp b/Applications/TransferLearning/CIFAR_Classification/jni/main.cpp
index d61fc6165f12f4955189f4ad61b8cbc07d6169dc..9bd27000d2cbf892644a871ed050bc3e4efe0cb1 100644 (file)
--- a/Applications/TransferLearning/CIFAR_Classification/jni/main.cpp
+++ b/Applications/TransferLearning/CIFAR_Classification/jni/main.cpp
@@ -416,15 +416,20 @@ int main(int argc, char *argv[]) {
     NN.initialize();
   } catch (...) {
     std::cerr << "Error during init" << std::endl;
-    return 0;
+    return 1;
+  }
+  try {
+    NN.readModel();
+  } catch (...) {
+    std::cerr << "Error during readModel" << std::endl;
+    return 1;
   }
-  NN.readModel();
 
   try {
     NN.train();
   } catch (...) {
     std::cerr << "Error during train" << std::endl;
-    return 0;
+    return 1;
   }
 
   if (!TRAINING) {

diff --git a/Applications/TransferLearning/CIFAR_Classification/jni/main_func.cpp b/Applications/TransferLearning/CIFAR_Classification/jni/main_func.cpp
index 628b6f86afbffce9096a65afdc3ead1474ac635f..faab06868513565f6fa55fba344f8a4d4891df87 100644 (file)
--- a/Applications/TransferLearning/CIFAR_Classification/jni/main_func.cpp
+++ b/Applications/TransferLearning/CIFAR_Classification/jni/main_func.cpp
@@ -289,7 +289,12 @@ int main(int argc, char *argv[]) {
     std::cerr << "Error during init" << std::endl;
     return 0;
   }
-  model->readModel();
+  try {
+    model->readModel();
+  } catch (...) {
+    std::cerr << "Error during readModel" << std::endl;
+    return 0;
+  }
   model->setDataset(dataset);
 
   /**