Skip to content

Commit

Permalink
init tensorrt_yolox
Browse files Browse the repository at this point in the history
Signed-off-by: Sena <[email protected]>
  • Loading branch information
LeoDriveProject authored and senagolcuk committed Mar 12, 2024
1 parent 595a908 commit a8de0dc
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions perception/tensorrt_yolox/src/tensorrt_yolox_node.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -41,9 +41,10 @@ TrtYoloXNode::TrtYoloXNode(const rclcpp::NodeOptions & node_options)

std::string model_path =
declare_parameter_with_description("model_path", "", "The onnx file name for YOLOX model");
std::string label_path = declare_parameter_with_description(
"label_path", "",
"The label file that consists of label name texts for detected object categories");
// std::string label_path = declare_parameter_with_description(
// "label_path", "",
// "The label file that consists of label name texts for detected object categories");
std::string label_path = "/home/golf/autoware_data/tensorrt_yolox/label.txt";
std::string precision = declare_parameter_with_description(
"precision", "fp32",
"operation precision to be used on inference. Valid value is one of: [fp32, fp16, int8]");
Expand Down

0 comments on commit a8de0dc

Please sign in to comment.