mmcv.ops.diff_iou_rotated_2d¶
- mmcv.ops.diff_iou_rotated_2d(box1: torch.Tensor, box2: torch.Tensor) → torch.Tensor[source]¶
Calculate differentiable iou of rotated 2d boxes.
- Parameters
box1 (Tensor) – (B, N, 5) First box.
box2 (Tensor) – (B, N, 5) Second box.
- Returns
(B, N) IoU.
- Return type
Tensor