diff options
| author | Stanislaw Halik <sthalik@misaki.pl> | 2021-05-05 20:42:54 +0200 |
|---|---|---|
| committer | Stanislaw Halik <sthalik@misaki.pl> | 2021-05-05 20:42:54 +0200 |
| commit | 3105083db6fad81d46de7291129e13a642e33ccf (patch) | |
| tree | d41a23457ec453f2959d1705080d1cd8cdea4090 /tracker-neuralnet/CMakeLists.txt | |
| parent | 0ddcf4eb94fa4ee530702bca2c54e390b3c6057d (diff) | |
| parent | c651d84eaa5f295a45e95edab28db9ca05013fe7 (diff) | |
Merge remote-tracking branch 'origin/unstable'
Diffstat (limited to 'tracker-neuralnet/CMakeLists.txt')
| -rw-r--r-- | tracker-neuralnet/CMakeLists.txt | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/tracker-neuralnet/CMakeLists.txt b/tracker-neuralnet/CMakeLists.txt new file mode 100644 index 00000000..d06f0c83 --- /dev/null +++ b/tracker-neuralnet/CMakeLists.txt @@ -0,0 +1,23 @@ +include(opentrack-opencv) +find_package(OpenCV QUIET) +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") + + 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) + + install( + FILES "models/head-localizer.onnx" "models/head-pose.onnx" + DESTINATION "${opentrack-libexec}/models" + PERMISSIONS ${opentrack-perms-file}) +endif()
\ No newline at end of file |