Skip to content

Commit

Permalink
fix(autoware_tensorrt_yolox): fix variableScope (#8430)
Browse files Browse the repository at this point in the history
fix: variableScope

Signed-off-by: Nagi70 <[email protected]>
Co-authored-by: kobayu858 <[email protected]>
  • Loading branch information
Nagi70 and kobayu858 authored Aug 15, 2024
1 parent 59a04e5 commit 36f30d4
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions perception/autoware_tensorrt_yolox/src/tensorrt_yolox.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -772,8 +772,6 @@ void TrtYoloX::multiScalePreprocess(const cv::Mat & image, const std::vector<cv:
bool TrtYoloX::doInferenceWithRoi(
const std::vector<cv::Mat> & images, ObjectArrays & objects, const std::vector<cv::Rect> & rois)
{
std::vector<cv::Mat> masks;
std::vector<cv::Mat> color_masks;
if (!trt_common_->isInitialized()) {
return false;
}
Expand All @@ -784,6 +782,8 @@ bool TrtYoloX::doInferenceWithRoi(
}

if (needs_output_decode_) {
std::vector<cv::Mat> masks;
std::vector<cv::Mat> color_masks;
return feedforwardAndDecode(images, objects, masks, color_masks);
} else {
return feedforward(images, objects);
Expand Down

0 comments on commit 36f30d4

Please sign in to comment.