Use correct traffic signs model

WATonomous · Jun 4, 2024 · 7e4aaa3 · 7e4aaa3
1 parent 4882626
commit 7e4aaa3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/modules/docker-compose.perception.yaml b/modules/docker-compose.perception.yaml
@@ -29,11 +29,11 @@ services:
             - driver: nvidia
               count: 1
               capabilities: [ gpu ]
-    command: /bin/bash -c "ros2 launch camera_object_detection eve_launch.py"
+    command: /bin/bash -c "ros2 launch camera_object_detection eve.launch.py"
     volumes:
       - /mnt/wato-drive2/perception_models/yolov8m.pt:/perception_models/yolov8m.pt
       - /mnt/wato-drive2/perception_models/traffic_light.pt:/perception_models/traffic_light.pt
-      - /mnt/wato-drive2/perception_models/traffic_signs_v0.pt:/perception_models/traffic_signs_v1.pt
+      - /mnt/wato-drive2/perception_models/traffic_signs_v3.pt:/perception_models/traffic_signs.pt
 
   lidar_object_detection:
     build:

diff --git a/src/perception/camera_object_detection/config/traffic_signs_config.yaml b/src/perception/camera_object_detection/config/traffic_signs_config.yaml
@@ -3,7 +3,7 @@ traffic_signs_node:
     camera_topic: /camera/left/image_color
     publish_vis_topic: /traffic_signs_viz
     publish_detection_topic: /traffic_signs
-    model_path: /perception_models/traffic_signs_v1.pt
+    model_path: /perception_models/traffic_signs.pt
     crop_mode: CenterCrop
     image_size: 1024
     save_detections: false