Skip to content

Commit

Permalink
feat(tensorrt_yolox): add yolox launch (autowarefoundation#2598)
Browse files Browse the repository at this point in the history
* feat(tensorrt_yolox): add yolox launch

Signed-off-by: Shunsuke Miura <shunsuke.miura@tier4.jp>

* add use_decompress option

Signed-off-by: Shunsuke Miura <shunsuke.miura@tier4.jp>

* ci(pre-commit): autofix

* Apply suggestions from code review

changes the topic name

Co-authored-by: Daisuke Nishimatsu <42202095+wep21@users.noreply.github.com>

Signed-off-by: Shunsuke Miura <shunsuke.miura@tier4.jp>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Daisuke Nishimatsu <42202095+wep21@users.noreply.github.com>
  • Loading branch information
3 people committed Jan 8, 2023
1 parent f1de244 commit 9b8d079
Show file tree
Hide file tree
Showing 2 changed files with 27 additions and 0 deletions.
1 change: 1 addition & 0 deletions perception/tensorrt_yolox/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -125,5 +125,6 @@ if(BUILD_TESTING)
endif()

ament_auto_package(INSTALL_TO_SHARE
launch
data
)
26 changes: 26 additions & 0 deletions perception/tensorrt_yolox/launch/yolox.launch.xml
Original file line number Diff line number Diff line change
@@ -0,0 +1,26 @@
<?xml version="1.0"?>
<launch>
<arg name="input/image" default="/sensing/camera/camera0/image_rect_color"/>
<arg name="output/objects" default="/perception/object_recognition/detection/rois0"/>
<arg name="model_name" default="yolox-tiny"/>
<arg name="model_path" default="$(find-pkg-share tensorrt_yolox)/data"/>
<arg name="score_threshold" default="0.35"/>
<arg name="nms_threshold" default="0.7"/>
<arg name="use_decompress" default="true" description="use image decompress"/>

<node pkg="image_transport_decompressor" exec="image_transport_decompressor_node" name="image_transport_decompressor_node" if="$(var use_decompress)">
<remap from="~/input/compressed_image" to="$(var input/image)/compressed"/>
<remap from="~/output/raw_image" to="$(var input/image)"/>
</node>

<node pkg="tensorrt_yolox" exec="tensorrt_yolox_node_exe" name="tensorrt_yolox" output="screen">
<remap from="~/in/image" to="$(var input/image)"/>
<remap from="~/out/objects" to="$(var output/objects)"/>
<param name="score_threshold" value="$(var score_threshold)"/>
<param name="model_path" value="$(var model_path)/$(var model_name).onnx"/>
<param name="label_path" value="$(var model_path)/label.txt"/>
<param name="trt_precision" value="fp16"/>
<param name="score_threshold" value="$(var score_threshold)"/>
<param name="nms_threshold" value="$(var nms_threshold)"/>
</node>
</launch>

0 comments on commit 9b8d079

Please sign in to comment.