Add lane_detection_tensorrt submodule

WATonomous · Mar 11, 2024 · 0211263 · 0211263
1 parent df8715a
commit 0211263
Show file tree

Hide file tree

Showing 40 changed files with 6 additions and 7,536 deletions.
diff --git a/modules/docker-compose.perception.yaml b/modules/docker-compose.perception.yaml
@@ -75,7 +75,7 @@ services:
               count: 1
               capabilities: [gpu]
     volumes:
-      - /mnt/wato-drive2/perception-weights/tensorrt_model/ultra_fast_lane_detection_v2/resource/model/ufldv2_culane_res34_320x1600.onnx:/home/bolty/ament_ws/build/lane_detection/resource/model/ufldv2_culane_res34_320x1600.onnx
+      - /mnt/wato-drive2/perception-weights/tensorrt_model/ultra_fast_lane_detection_v2/resource/model/ufldv2_culane_res34_320x1600.onnx:/models/ufldv2_culane_res34_320x1600.onnx
 
   tracking:
     build:

diff --git a/src/perception/lane_detection/CMakeLists.txt b/src/perception/lane_detection/CMakeLists.txt
@@ -17,20 +17,20 @@ endif()
 ### TensorRT model ###
 
 # Select build system and set compile options
-include(${CMAKE_CURRENT_LIST_DIR}/src/tensorrt_model/common_helper/cmakes/build_setting.cmake)
+include(${CMAKE_CURRENT_LIST_DIR}/src/lane_detection_tensorrt/common_helper/cmakes/build_setting.cmake)
 
 add_executable(lane_detection src/lane_detection_node.cpp)
 
-add_subdirectory(./src/tensorrt_model/ultra_fast_lane_detection_v2/image_processor image_processor)
-target_include_directories(${ProjectName} PUBLIC ./src/tensorrt_model/ultra_fast_lane_detection_v2/image_processor)
+add_subdirectory(./src/lane_detection_tensorrt/ultra_fast_lane_detection_v2/image_processor image_processor)
+target_include_directories(${ProjectName} PUBLIC ./src/lane_detection_tensorrt/ultra_fast_lane_detection_v2/image_processor)
 target_link_libraries(${ProjectName} ImageProcessor)
 
 find_package(OpenCV REQUIRED)
 target_include_directories(${ProjectName} PUBLIC ${OpenCV_INCLUDE_DIRS})
 target_link_libraries(${ProjectName} ${OpenCV_LIBS})
 
 # Copy resouce
-add_definitions(-DRESOURCE_DIR="${CMAKE_BINARY_DIR}/resource/")
+add_definitions(-DRESOURCE_DIR="/models/")
 
 ### End of TensorRT model ###
 

diff --git a/src/perception/lane_detection/src/lane_detection_tensorrt b/src/perception/lane_detection/src/lane_detection_tensorrt
diff --git a/src/perception/lane_detection/src/tensorrt_model/InferenceHelper/LICENSE b/src/perception/lane_detection/src/tensorrt_model/InferenceHelper/LICENSE