Skip to content
This repository has been archived by the owner on Mar 27, 2023. It is now read-only.

Commit

Permalink
chore: expose traffic light model path arg (#225)
Browse files Browse the repository at this point in the history
Signed-off-by: wep21 <border_goldenmarket@yahoo.co.jp>
  • Loading branch information
wep21 committed Mar 2, 2022
1 parent c2fed61 commit d1edfd6
Showing 1 changed file with 16 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,18 @@
<arg name="enable_fine_detection" default="true" description="enable fine position adjustment of traffic light"/>
<arg name="input/image" default="/sensing/camera/traffic_light/image_raw" description="image raw topic name for traffic light"/>
<arg name="input/camera_info" default="/sensing/camera/traffic_light/camera_info" description="camera info topic name for traffic light"/>
<arg name="fine_detector_label_path"
default="$(find-pkg-share traffic_light_ssd_fine_detector)/data/voc_labels_tl.txt"
description="fine detector label path"/>
<arg name="fine_detector_model_path"
default="$(find-pkg-share traffic_light_ssd_fine_detector)/data/mb2-ssd-lite-tlr.onnx"
description="fine detector onnx model path"/>
<arg name="classifier_label_path"
default="$(find-pkg-share traffic_light_classifier)/data/lamp_labels.txt"
description="classifier label path"/>
<arg name="classifier_model_path"
default="$(find-pkg-share traffic_light_classifier)/data/traffic_light_classifier_mobilenetv2.onnx"
description="classifier onnx model path"/>

<group if="$(var enable_fine_detection)">
<include file="$(find-pkg-share traffic_light_map_based_detector)/launch/traffic_light_map_based_detector.launch.xml">
Expand All @@ -27,6 +39,10 @@
<arg name="enable_fine_detection" value="$(var enable_fine_detection)"/>
<arg name="use_intra_process" value="true"/>
<arg name="use_multithread" value="false"/>
<arg name="label_file" value="$(var fine_detector_label_path)"/>
<arg name="onnx_file" value="$(var fine_detector_model_path)"/>
<arg name="label_file_path" value="$(var classifier_label_path)"/>
<arg name="model_file_path" value="$(var classifier_model_path)"/>
</include>

<!-- visualizer -->
Expand Down

0 comments on commit d1edfd6

Please sign in to comment.