From 1325889305595f4def85fef61ef1f2082cfc441b Mon Sep 17 00:00:00 2001 From: higor-melo <115033428+higor-melo@users.noreply.github.com> Date: Wed, 27 Mar 2024 15:31:42 -0300 Subject: [PATCH] Fix typo in YOLOv8-Libtorch-CPP-Inference (#9330) Co-authored-by: Glenn Jocher --- examples/YOLOv8-LibTorch-CPP-Inference/main.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/YOLOv8-LibTorch-CPP-Inference/main.cc b/examples/YOLOv8-LibTorch-CPP-Inference/main.cc index ebb1a758..b68b7f7e 100644 --- a/examples/YOLOv8-LibTorch-CPP-Inference/main.cc +++ b/examples/YOLOv8-LibTorch-CPP-Inference/main.cc @@ -139,7 +139,7 @@ torch::Tensor nms(const torch::Tensor& bboxes, const torch::Tensor& scores, floa } -torch::Tensor non_max_supperession(torch::Tensor& prediction, float conf_thres = 0.25, float iou_thres = 0.45, int max_det = 300) { +torch::Tensor non_max_suppression(torch::Tensor& prediction, float conf_thres = 0.25, float iou_thres = 0.45, int max_det = 300) { auto bs = prediction.size(0); auto nc = prediction.size(1) - 4; auto nm = prediction.size(1) - nc - 4; @@ -237,7 +237,7 @@ int main() { torch::Tensor output = yolo_model.forward(inputs).toTensor().cpu(); // NMS - auto keep = non_max_supperession(output)[0]; + auto keep = non_max_suppression(output)[0]; auto boxes = keep.index({Slice(), Slice(None, 4)}); keep.index_put_({Slice(), Slice(None, 4)}, scale_boxes({input_image.rows, input_image.cols}, boxes, {image.rows, image.cols}));