deep_learning/ultralytics
1
0
Fork 0
mirror of https://github.com/ultralytics/ultralytics.git synced 2025-09-15 15:48:41 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Fix scaling ratio calculation in ONNX example (#20016)

Browse Source 
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
This commit is contained in:
MrBlackBigWhiteSheep 2025-04-05 07:32:47 +08:00 committed by GitHub
parent d610180309
commit 608617e598
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/YOLOv8-ONNXRuntime/main.py
View File

@ -181,7 +181,7 @@ class YOLOv8:
class_ids = []
# Calculate the scaling factors for the bounding box coordinates
gain = min(self.input_height / self.img_height, self.input_width / self.img_height)
gain = min(self.input_height / self.img_height, self.input_width / self.img_width)
outputs[:, 0] -= pad[1]
outputs[:, 1] -= pad[0]