forked from dsutanto/bytetrack-counter-cpp
103 lines
2.3 KiB
CMake
103 lines
2.3 KiB
CMake
cmake_minimum_required(VERSION 3.16)
|
|
project(bytetrack-counter LANGUAGES CXX)
|
|
|
|
set(CMAKE_CXX_STANDARD 17)
|
|
set(CMAKE_CXX_STANDARD_REQUIRED ON)
|
|
set(CMAKE_CXX_EXTENSIONS OFF)
|
|
|
|
find_package(OpenCV REQUIRED COMPONENTS core imgproc videoio highgui imgcodecs)
|
|
set(Eigen3_DIR /usr/share/eigen3/cmake)
|
|
find_package(Eigen3 REQUIRED NO_MODULE)
|
|
find_package(SQLite3 REQUIRED)
|
|
|
|
option(HAS_RKNN "Build with RKNN NPU support" ON)
|
|
|
|
if(HAS_RKNN)
|
|
add_definitions(-DHAS_RKNN)
|
|
include_directories(/usr/local/include/rknn)
|
|
set(HAS_RKNN_VAL 1)
|
|
else()
|
|
set(HAS_RKNN_VAL 0)
|
|
endif()
|
|
|
|
execute_process(
|
|
COMMAND git rev-parse --short=8 HEAD
|
|
WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}
|
|
OUTPUT_VARIABLE GIT_COMMIT
|
|
OUTPUT_STRIP_TRAILING_WHITESPACE
|
|
ERROR_QUIET
|
|
)
|
|
if(NOT GIT_COMMIT)
|
|
set(GIT_COMMIT "unknown")
|
|
endif()
|
|
|
|
execute_process(
|
|
COMMAND git rev-parse --abbrev-ref HEAD
|
|
WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}
|
|
OUTPUT_VARIABLE GIT_BRANCH
|
|
OUTPUT_STRIP_TRAILING_WHITESPACE
|
|
ERROR_QUIET
|
|
)
|
|
if(NOT GIT_BRANCH)
|
|
set(GIT_BRANCH "unknown")
|
|
endif()
|
|
|
|
execute_process(
|
|
COMMAND git describe --tags --exact-match 2>/dev/null || echo ""
|
|
WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}
|
|
OUTPUT_VARIABLE GIT_TAG
|
|
OUTPUT_STRIP_TRAILING_WHITESPACE
|
|
ERROR_QUIET
|
|
)
|
|
if(NOT GIT_TAG)
|
|
set(GIT_TAG "")
|
|
endif()
|
|
|
|
string(TIMESTAMP BUILD_DATE "%Y-%m-%d")
|
|
string(TIMESTAMP BUILD_TIME "%H:%M:%S")
|
|
string(TIMESTAMP BUILD_YEAR "%Y")
|
|
|
|
configure_file(
|
|
${CMAKE_SOURCE_DIR}/src/version.hpp.in
|
|
${CMAKE_BINARY_DIR}/version.hpp
|
|
@ONLY
|
|
)
|
|
|
|
set(SOURCES
|
|
src/main.cpp
|
|
src/kalman_filter.cpp
|
|
src/bytetrack.cpp
|
|
src/batch_store.cpp
|
|
src/csv_logger.cpp
|
|
src/rknn_yolo.cpp
|
|
src/config.cpp
|
|
src/drawing.cpp
|
|
)
|
|
|
|
add_executable(bytetrack-counter ${SOURCES})
|
|
|
|
target_include_directories(bytetrack-counter PRIVATE
|
|
${OpenCV_INCLUDE_DIRS}
|
|
${EIGEN3_INCLUDE_DIR}
|
|
${SQLite3_INCLUDE_DIRS}
|
|
src
|
|
${CMAKE_BINARY_DIR}
|
|
)
|
|
|
|
target_link_libraries(bytetrack-counter
|
|
${OpenCV_LIBS}
|
|
${SQLite3_LIBRARIES}
|
|
pthread
|
|
)
|
|
|
|
if(HAS_RKNN)
|
|
find_library(RKNN_LIB rknnrt PATHS /usr/lib /usr/local/lib)
|
|
if(RKNN_LIB)
|
|
target_link_libraries(bytetrack-counter ${RKNN_LIB})
|
|
else()
|
|
message(WARNING "librknnrt not found — RKNN support disabled")
|
|
endif()
|
|
endif()
|
|
|
|
install(TARGETS bytetrack-counter DESTINATION /opt/bytetrack-counter-cpp/bin)
|