diff options
| author | Stanislaw Halik <sthalik@misaki.pl> | 2021-08-03 08:04:30 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-08-03 08:04:30 +0000 |
| commit | 6a5954197ef5671b5cc9159377e83ef15bf43373 (patch) | |
| tree | c4ea500f2b42636464b844bd831b6382f931e0ba /tracker-neuralnet/CMakeLists.txt | |
| parent | fef1936f3c320ec026a0fdf65f103494385e15e3 (diff) | |
| parent | 1e3d62e04f78e2d0e22b3ceb67aaf20a03f18bbc (diff) | |
Merge pull request #1306 from gdolle/tracker-neuralnet-cmake
Tracker neuralnet update onnxrt cmake (linux)
Diffstat (limited to 'tracker-neuralnet/CMakeLists.txt')
| -rw-r--r-- | tracker-neuralnet/CMakeLists.txt | 29 |
1 files changed, 13 insertions, 16 deletions
diff --git a/tracker-neuralnet/CMakeLists.txt b/tracker-neuralnet/CMakeLists.txt index d06f0c83..bcdbbc81 100644 --- a/tracker-neuralnet/CMakeLists.txt +++ b/tracker-neuralnet/CMakeLists.txt @@ -1,23 +1,20 @@ include(opentrack-opencv) -find_package(OpenCV QUIET) +find_package(OpenCV QUIET COMPONENTS imgproc core imgcodecs calib3d) find_package(OpenMP QUIET) # Used to control number of onnx threads. -set(SDK_ONNX_LIBPATH "" CACHE FILEPATH "Full path of onnx library") - -if(OpenCV_FOUND AND SDK_ONNX_LIBPATH AND OpenMP_FOUND) - get_filename_component(ONNX_INCLUDE_DIR "${SDK_ONNX_LIBPATH}" DIRECTORY) - get_filename_component(ONNX_INCLUDE_DIR "${ONNX_INCLUDE_DIR}" ABSOLUTE) - set(ONNX_INCLUDE_DIR "${ONNX_INCLUDE_DIR}/../include") +find_package(ONNXRuntime QUIET) +if(OpenCV_FOUND AND ONNXRuntime_FOUND AND OpenMP_FOUND) otr_module(tracker-neuralnet) - target_include_directories(${self} SYSTEM PUBLIC - ${OpenCV_INCLUDE_DIRS} "${ONNX_INCLUDE_DIR}") target_link_libraries(${self} - opentrack-cv "${SDK_ONNX_LIBPATH}" opencv_imgproc opencv_core - opencv_imgcodecs opencv_calib3d - OpenMP::OpenMP_C) - + opentrack-cv + onnxruntime + ${OpenCV_LIBS} + OpenMP::OpenMP_C + ) install( - FILES "models/head-localizer.onnx" "models/head-pose.onnx" + FILES "models/head-localizer.onnx" + "models/head-pose.onnx" DESTINATION "${opentrack-libexec}/models" - PERMISSIONS ${opentrack-perms-file}) -endif()
\ No newline at end of file + PERMISSIONS ${opentrack-perms-file} + ) +endif() |