Fix OBB xywh-angle concatenation bug (#7404)

2024-01-09 01:27:43 +08:00 · 2024-01-09 01:27:43 +08:00 · e19398a537
commit e19398a537
parent ebf8dc0558
2 changed files with 2 additions and 2 deletions
--- a/ultralytics/utils/ops.py
+++ b/ultralytics/utils/ops.py
@ -263,7 +263,7 @@ def non_max_suppression(
        c = x[:, 5:6] * (0 if agnostic else max_wh)  # classes
        scores = x[:, 4]  # scores
        if rotated:
-            boxes = torch.cat((x[:, :2] + c, x[:, 2:4], x[:, -2:-1]), dim=-1)  # xywhr
+            boxes = torch.cat((x[:, :2] + c, x[:, 2:4], x[:, -1:]), dim=-1)  # xywhr
            i = nms_rotated(boxes, scores, iou_thres)
        else:
            boxes = x[:, :4] + c  # boxes (offset by class)