first commit

third_party/trt_vehicle/CMakeLists.txt

@@ -0,0 +1,72 @@
+cmake_minimum_required(VERSION 3.10)
+project(trt_vehicle VERSION 1.0)
+
+set(CMAKE_CXX_STANDARD 17)
+set(CMAKE_BUILD_TYPE Debug)
+set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -g -fPIC -w -fdiagnostics-color=always -pthread")
+# save all libs to the same directory
+set(CMAKE_LIBRARY_OUTPUT_DIRECTORY ${CMAKE_BINARY_DIR}/libs)
+
+# TensorRT required
+set(TRT_LIB_PATH "/usr/local/tensorRT/lib")      # change this line if possible
+set(TRT_INC_PATH "/usr/local/tensorRT/include")  # change this line if possible
+link_directories(${TRT_LIB_PATH})     
+include_directories(${TRT_INC_PATH})
+if(DEFINED VP_BUILD_FROM) # tell videopipe where is trt
+    set(VP_TRT_LIB_PATH ${TRT_LIB_PATH} PARENT_SCOPE)     
+    set(VP_TRT_INC_PATH ${TRT_INC_PATH} PARENT_SCOPE) 
+endif()
+
+# CUDA required
+find_package(CUDA REQUIRED) 
+message(STATUS "CUDA library status:")
+message(STATUS "    version: ${CUDA_VERSION}")
+message(STATUS "    libraries: ${CUDA_LIBRARIES}")
+message(STATUS "    include path: ${CUDA_INCLUDE_DIRS}")
+include_directories(${CUDA_INCLUDE_DIRS})
+
+# OpenCV required
+find_package(OpenCV REQUIRED)
+message(STATUS "OpenCV library status:")
+message(STATUS "    version: ${OpenCV_VERSION}")
+message(STATUS "    libraries: ${OpenCV_LIBS}")
+message(STATUS "    include path: ${OpenCV_INCLUDE_DIRS}")
+include_directories(${OpenCV_INCLUDE_DIRS})
+
+# collect dependent libs for trt_vehicle
+set(TRT_VEHICLE_DEPEND_LIBS ${OpenCV_LIBS} ${CUDA_LIBRARIES} nvinfer stdc++fs)
+
+# collect source files for trt_vehicle
+file(GLOB_RECURSE MODELS "models/*.cpp")
+file(GLOB_RECURSE UTIL "util/*.cpp")
+#...#
+list(APPEND TRT_VEHICLE_CPPS ${MODELS} ${UTIL})
+
+# build for trt_vehicle
+add_library(${PROJECT_NAME} SHARED ${TRT_VEHICLE_CPPS})
+target_link_libraries(${PROJECT_NAME} ${TRT_VEHICLE_DEPEND_LIBS})
+
+# build samples for trt_vehicle
+if(NOT DEFINED VP_BUILD_FROM)
+    set(CMAKE_RUNTIME_OUTPUT_DIRECTORY ${CMAKE_BINARY_DIR}/samples)  # save all exe to 'samples'
+    add_executable(vehicle_cluster "samples/vehicle_cluster.cpp")
+    target_link_libraries(vehicle_cluster ${PROJECT_NAME})
+
+    add_executable(vehicle_color_type "samples/vehicle_color_type.cpp")
+    target_link_libraries(vehicle_color_type ${PROJECT_NAME})
+
+    add_executable(vehicle_compare "samples/vehicle_compare.cpp")
+    target_link_libraries(vehicle_compare ${PROJECT_NAME})
+
+    add_executable(vehicle_plate "samples/vehicle_plate.cpp")
+    target_link_libraries(vehicle_plate ${PROJECT_NAME})
+
+    add_executable(vehicle_scan "samples/vehicle_scan.cpp")
+    target_link_libraries(vehicle_scan ${PROJECT_NAME})
+
+    add_executable(vehicle_search "samples/vehicle_search.cpp")
+    target_link_libraries(vehicle_search ${PROJECT_NAME})
+
+    add_executable(vehicle "samples/vehicle.cpp")
+    target_link_libraries(vehicle ${PROJECT_NAME})
+endif()