mirror of
https://github.com/ok-oldking/ok-wuthering-waves.git
synced 2025-04-24 08:25:16 +00:00
fix yolo
This commit is contained in:
parent
1a3d9f4977
commit
6a2570ea8d
@ -123,7 +123,7 @@ class OpenVinoYolo8Detect: # Renamed class
|
||||
class_ids = []
|
||||
|
||||
# Calculate the scaling factors for the bounding box coordinates
|
||||
gain = min(orig_shape[0] / orig_shape[0], self.input_width / orig_shape[1])
|
||||
gain = min(self.input_height / orig_shape[0], self.input_width / orig_shape[1])
|
||||
|
||||
outputs[:, 0] -= padding[1]
|
||||
outputs[:, 1] -= padding[0]
|
||||
|
Loading…
x
Reference in New Issue
Block a user