| | |
| | from __future__ import absolute_import, division, print_function, unicode_literals |
| | import torch |
| |
|
| |
|
| | def pairwise_iou_rotated(boxes1, boxes2): |
| | """ |
| | Return intersection-over-union (Jaccard index) of boxes. |
| | |
| | Both sets of boxes are expected to be in |
| | (x_center, y_center, width, height, angle) format. |
| | |
| | Arguments: |
| | boxes1 (Tensor[N, 5]) |
| | boxes2 (Tensor[M, 5]) |
| | |
| | Returns: |
| | iou (Tensor[N, M]): the NxM matrix containing the pairwise |
| | IoU values for every element in boxes1 and boxes2 |
| | """ |
| | return torch.ops.detectron2.box_iou_rotated(boxes1, boxes2) |
| |
|
