diff options
Diffstat (limited to 'runtime/contrib/tflite_classify/CMakeLists.txt')
| -rw-r--r-- | runtime/contrib/tflite_classify/CMakeLists.txt | 22 |
1 files changed, 22 insertions, 0 deletions
diff --git a/runtime/contrib/tflite_classify/CMakeLists.txt b/runtime/contrib/tflite_classify/CMakeLists.txt new file mode 100644 index 000000000..c0bf62738 --- /dev/null +++ b/runtime/contrib/tflite_classify/CMakeLists.txt @@ -0,0 +1,22 @@ +if(NOT BUILD_TFLITE_CLASSIFY_APP) + return() +endif(NOT BUILD_TFLITE_CLASSIFY_APP) + +list(APPEND SOURCES "src/tflite_classify.cc") +list(APPEND SOURCES "src/ImageClassifier.cc") +list(APPEND SOURCES "src/InferenceInterface.cc") + +## Required package +find_package(OpenCV REQUIRED) +find_package(Boost REQUIRED COMPONENTS system filesystem) + +# Without this line, this appliation couldn't search the opencv library that were already installed in ${ROOTFS_DIR}/usr/lib/arm-linux-gnueabihf directory +set(CMAKE_EXE_LINKER_FLAGS "-Wl,--as-needed -Wl,--rpath=${ROOTFS_DIR}/usr/lib/arm-linux-gnueabihf -Wl,--rpath=${ROOTFS_DIR}/lib/arm-linux-gnueabihf") + +add_executable(tflite_classify ${SOURCES}) +target_include_directories(tflite_classify PRIVATE src) +target_link_libraries(tflite_classify tensorflow-lite ${LIB_PTHREAD} dl nnfw_lib_tflite) +target_link_libraries(tflite_classify ${Boost_LIBRARIES}) +target_link_libraries(tflite_classify ${OpenCV_LIBRARIES}) + +install(TARGETS tflite_classify DESTINATION bin) |