Skip to content

Commit

Permalink
Update rotate_iou2d_calculator.py
Browse files Browse the repository at this point in the history
  • Loading branch information
zytx121 committed Sep 9, 2022
1 parent a02ca62 commit 11cf17f
Showing 1 changed file with 14 additions and 10 deletions.
24 changes: 14 additions & 10 deletions mmrotate/core/bbox/iou_calculators/rotate_iou2d_calculator.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,10 +20,12 @@ def __call__(self,
Args:
bboxes1 (:obj:`RotatedBoxes` or Tensor): bboxes have shape (m, 5)
in <cx, cy, w, h, t> format, shape (m, 6) in
<cx, cy, w, h, t, score> format.
bboxes2 (:obj:`RotatedBoxes` or Tensor): bboxes have shape (m, 5)
in <cx, cy, w, h, t> format, shape (m, 6) in
<cx, cy, w, h, t, score> format, or be empty.
<cx, cy, w, h, t, score> format. 't' is the box angle
represented in radian.
bboxes2 (:obj:`RotatedBoxes` or Tensor): bboxes have shape (n, 5)
in <cx, cy, w, h, t> format, shape (n, 6) in
<cx, cy, w, h, t, score> format, or be empty. 't' is the box
angle represented in radian.
mode (str): 'iou' (intersection over union), 'iof' (intersection
over foreground). Defaults to 'iou'.
is_aligned (bool): If True, then m and n must be equal.
Expand Down Expand Up @@ -60,9 +62,9 @@ def rbbox_overlaps(bboxes1: Tensor,
"""Calculate overlap between two set of rotated bboxes.
Args:
bboxes1 (Tensor): shape (B, m, 5) in <cx, cy, w, h, a> format
bboxes1 (Tensor): shape (B, m, 5) in <cx, cy, w, h, t> format
or empty.
bboxes2 (Tensor): shape (B, n, 5) in <cx, cy, w, h, a> format
bboxes2 (Tensor): shape (B, n, 5) in <cx, cy, w, h, t> format
or empty.
mode (str): 'iou' (intersection over union), 'iof' (intersection over
foreground). Defaults to 'iou'.
Expand Down Expand Up @@ -109,10 +111,12 @@ def __call__(self,
Args:
bboxes1 (:obj:`RotatedBoxes` or Tensor): bboxes have shape (m, 5)
in <cx, cy, w, h, t> format, shape (m, 6) in
<cx, cy, w, h, t, score> format.
bboxes2 (:obj:`RotatedBoxes` or Tensor): bboxes have shape (m, 5)
in <cx, cy, w, h, t> format, shape (m, 6) in
<cx, cy, w, h, t, score> format, or be empty.
<cx, cy, w, h, t, score> format. 't' is the box angle
represented in radian.
bboxes2 (:obj:`RotatedBoxes` or Tensor): bboxes have shape (n, 5)
in <cx, cy, w, h, t> format, shape (n, 6) in
<cx, cy, w, h, t, score> format, or be empty. 't' is the box
angle represented in radian.
mode (str): 'iou' (intersection over union), 'iof' (intersection
over foreground).
is_aligned (bool): If True, then m and n must be equal.
Expand Down

0 comments on commit 11cf17f

Please sign in to comment.