diff options
| author | Michael Welter <michael@welter-4d.de> | 2021-03-13 11:27:03 +0100 |
|---|---|---|
| committer | Michael Welter <michael@welter-4d.de> | 2021-05-02 18:20:41 +0200 |
| commit | 7e582247ab0942398225b3e190d9fadafcf19950 (patch) | |
| tree | 394d8ca7c1aedf058208191b19b8e3e9fe4d134c /tracker-neuralnet/CMakeLists.txt | |
| parent | 058942f40e17e091b91df5436d771d61203ccc73 (diff) | |
Add a face alignment based tracker
Powered by AI!
Models were generated with code from
https://github.com/DaWelter/neuralnet-tracker-traincode/releases/tag/v0.1
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 |