Check if model file path is valid or not 73/233073/1
authorInki Dae <inki.dae@samsung.com>
Tue, 12 May 2020 04:02:31 +0000 (13:02 +0900)
committerInki Dae <inki.dae@samsung.com>
Tue, 12 May 2020 04:05:07 +0000 (13:05 +0900)
Change-Id: Ib1b2f3f897d4ee0d3ff4646429308ded762651ea
Signed-off-by: Inki Dae <inki.dae@samsung.com>
src/inference_engine_tflite.cpp

index 9316f1678b095ea5b92dc3954ca375a2a79787ca..bd44e06106a27a633dec6cad56fd8c02b4d5b853 100644 (file)
@@ -57,6 +57,11 @@ int InferenceTFLite::Load(std::vector<std::string> model_paths, inference_model_
 
     mWeightFile = model_paths.back();
 
+       if (access(mWeightFile.c_str(), F_OK)) {
+               LOGE("model file path [%s]", mWeightFile.c_str());
+               return INFERENCE_ENGINE_ERROR_INVALID_PATH;
+       }
+
     LOGI("mWeightFile.c_str() result [%s]", mWeightFile.c_str());
 
     mFlatBuffModel = tflite::FlatBufferModel::BuildFromFile(mWeightFile.c_str());