diff --git a/model_zoo/official/lite/image_classification/app/CMakeLists.txt b/model_zoo/official/lite/image_classification/app/CMakeLists.txt index 175b1f24c3..11ab1972a7 100644 --- a/model_zoo/official/lite/image_classification/app/CMakeLists.txt +++ b/model_zoo/official/lite/image_classification/app/CMakeLists.txt @@ -14,6 +14,7 @@ set(MINDSPORELITE_VERSION mindspore-lite-1.1.1-runtime-arm64-cpu) include_directories(${CMAKE_SOURCE_DIR}/src/main/cpp) include_directories(${CMAKE_SOURCE_DIR}/src/main/cpp/${MINDSPORELITE_VERSION}) include_directories(${CMAKE_SOURCE_DIR}/src/main/cpp/${MINDSPORELITE_VERSION}/include) +include_directories(${CMAKE_SOURCE_DIR}/src/main/cpp/${MINDSPORELITE_VERSION}/minddata/include) add_library(mindspore-lite SHARED IMPORTED) add_library(minddata-lite SHARED IMPORTED) diff --git a/model_zoo/official/lite/image_classification/app/src/main/cpp/MindSporeNetnative.cpp b/model_zoo/official/lite/image_classification/app/src/main/cpp/MindSporeNetnative.cpp index 6a684b5905..796aae398e 100644 --- a/model_zoo/official/lite/image_classification/app/src/main/cpp/MindSporeNetnative.cpp +++ b/model_zoo/official/lite/image_classification/app/src/main/cpp/MindSporeNetnative.cpp @@ -27,8 +27,8 @@ #include "include/ms_tensor.h" #include "MindSporeNetnative.h" #include "MSNetWork.h" -#include "minddata/include/lite_cv/lite_mat.h" -#include "minddata/include/lite_cv/image_process.h" +#include "lite_cv/lite_mat.h" +#include "lite_cv/image_process.h" using mindspore::dataset::LiteMat; using mindspore::dataset::LPixelType;