Commit 382be011 authored by Lozac'h Loic's avatar Lozac'h Loic
Browse files

Modifying cmake files

parent c7f8d0a4
......@@ -4,6 +4,23 @@ project(AgriSoilMoisture)
# This module create a library, with the same name as the module
set(AgriSoilMoisture_LIBRARIES AgriSoilMoisture)
if(OTB_USE_TENSORFLOW)
message("Tensorflow support enabled")
# find Tensorflow INCLUDE DIR
set(tensorflow_include_dir "" CACHE PATH "The include directory of tensorflow")
include_directories(${tensorflow_include_dir})
# find Tensorflow LIBRARIES
find_library(TENSORFLOW_CC_LIB NAMES libtensorflow_cc)
find_library(TENSORFLOW_FRAMEWORK_LIB NAMES libtensorflow_framework)
set(TENSORFLOW_LIBS "${TENSORFLOW_CC_LIB}" "${TENSORFLOW_FRAMEWORK_LIB}")
else()
message("Tensorflow support disabled")
endif()
if(NOT OTB_SOURCE_DIR)
# Handle the compilation outside OTB source tree: find where OTB is built/installed
find_package(OTB REQUIRED)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment