diff options
| author | Stanislaw Halik <sthalik@misaki.pl> | 2021-05-03 11:04:08 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-05-03 11:04:08 +0000 |
| commit | c651d84eaa5f295a45e95edab28db9ca05013fe7 (patch) | |
| tree | 394d8ca7c1aedf058208191b19b8e3e9fe4d134c /tracker-neuralnet/CMakeLists.txt | |
| parent | 058942f40e17e091b91df5436d771d61203ccc73 (diff) | |
| parent | 7e582247ab0942398225b3e190d9fadafcf19950 (diff) | |
Merge pull request #1266 from DaWelter/feature-tracker-neuralnet-prunstable
Add a face alignment based tracker
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 |