summaryrefslogtreecommitdiffhomepage
path: root/tracker-neuralnet/CMakeLists.txt
blob: c38ba3d8338f1801202000bd92a1993a86d173ff (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
include(opentrack-opencv)
find_package(OpenCV QUIET)
find_package(OpenMP QUIET) # Used to control number of onnx threads.
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} "${ONNXRuntime_INCLUDE_DIRS}")
    target_link_libraries(${self} "${ONNXRuntime_LIBRARIES}"
        opentrack-cv 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()