Skip to content

Commit

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

fix: variableScope

Signed-off-by: Nagi70 <n.yoshimura.060@ms.saitama-u.ac.jp>
Co-authored-by: kobayu858 <129580202+kobayu858@users.noreply.github.com>
Signed-off-by: xtk8532704 <1041084556@qq.com>
  • Loading branch information
2 people authored and xtk8532704 committed Aug 15, 2024
1 parent 8378d0f commit d1fe02e
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 d1fe02e

Please sign in to comment.