Fix cmake

iwatake2222 · Aug 21, 2020 · 2014342 · 2014342
1 parent ec49db4
commit 2014342
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/temp_pj_tflite_simple_cls_mobilenet_v2/ImageProcessor/CMakeLists.txt b/temp_pj_tflite_simple_cls_mobilenet_v2/ImageProcessor/CMakeLists.txt
@@ -11,6 +11,12 @@ target_include_directories(${LibraryName} PUBLIC ${OpenCV_INCLUDE_DIRS})
 target_link_libraries(${LibraryName} ${OpenCV_LIBS})
 
 # For Tensorflow Lite
-include(${CMAKE_CURRENT_LIST_DIR}/../../third_party/cmakes/tensorflow.cmake)
+include(${CMAKE_CURRENT_LIST_DIR}/../../third_party/cmakes/tflite.cmake)
 target_include_directories(${LibraryName} PUBLIC ${TFLITE_INC})
 target_link_libraries(${LibraryName} ${TFLITE_LIB})
+
+# For Tensorflow Lite Delegate(Edge TPU)
+include(${CMAKE_CURRENT_LIST_DIR}/../../third_party/cmakes/tflite_edgetpu.cmake)
+target_include_directories(${LibraryName} PUBLIC ${TFLITE_EDGETPI_INC})
+target_link_libraries(${LibraryName} ${TFLITE_EDGETPI_LIB})
+add_definitions(-DTFLITE_DELEGATE_EDGETPU)