Fix save_hybrid (#4245)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-08-10 00:51:21 +08:00
parent 7dfdb63cde
commit e9c9b82c42
2 changed files with 10 additions and 5 deletions
--- a/ultralytics/utils/ops.py
+++ b/ultralytics/utils/ops.py
@ -225,8 +225,8 @@ def non_max_suppression(
        # Cat apriori labels if autolabelling
        if labels and len(labels[xi]):
            lb = labels[xi]
-            v = torch.zeros((len(lb), nc + nm + 5), device=x.device)
-            v[:, :4] = lb[:, 1:5]  # box
+            v = torch.zeros((len(lb), nc + nm + 4), device=x.device)
+            v[:, :4] = xywh2xyxy(lb[:, 1:5])  # box
            v[range(len(lb)), lb[:, 0].long() + 4] = 1.0  # cls
            x = torch.cat((x, v), 0)