Elgato_dark/ultralytics
1
0
Fork 0
mirror of https://github.com/ultralytics/ultralytics synced 2026-04-21 14:07:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 8

fix openvino sample for coordinates (#23366)

Browse source
This commit is contained in:
Jules 2026-01-23 14:17:46 +08:00 committed by GitHub
parent ab579fa073
commit 4996590a3f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 8 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
examples/YOLOv8-OpenVINO-CPP-Inference/inference.cc
View file

@ -107,14 +107,14 @@ void Inference::PostProcessing(cv::Mat &frame) {
class_list.push_back(class_id.y);
confidence_list.push_back(score);
const float x = detection_outputs.at<float>(0, i);
const float y = detection_outputs.at<float>(1, i);
const float cx = detection_outputs.at<float>(0, i);
const float cy = detection_outputs.at<float>(1, i);
const float w = detection_outputs.at<float>(2, i);
const float h = detection_outputs.at<float>(3, i);
cv::Rect box;
box.x = static_cast<int>(x);
box.y = static_cast<int>(y);
box.x = static_cast<int>((cx - w / 2));
box.y = static_cast<int>((cy - h / 2));
box.width = static_cast<int>(w);
box.height = static_cast<int>(h);
box_list.push_back(box);
@ -141,10 +141,10 @@ void Inference::PostProcessing(cv::Mat &frame) {
// Method to get the bounding box in the correct scale
cv::Rect Inference::GetBoundingBox(const cv::Rect &src) const {
cv::Rect box = src;
box.x = (box.x - box.width / 2) * scale_factor_.x;
box.y = (box.y - box.height / 2) * scale_factor_.y;
box.width *= scale_factor_.x;
box.height *= scale_factor_.y;
box.x = static_cast<int>(box.x * scale_factor_.x);
box.y = static_cast<int>(box.y * scale_factor_.y);
box.width = static_cast<int>(box.width * scale_factor_.x);
box.height = static_cast<int>(box.height * scale_factor_.y);
return box;
}