mmcv.ops.diff_iou_rotated_2d¶
- mmcv.ops.diff_iou_rotated_2d(box1: torch.Tensor, box2: torch.Tensor) → torch.Tensor[源代码]¶
Calculate differentiable iou of rotated 2d boxes.
- 参数
box1 (Tensor) – (B, N, 5) First box.
box2 (Tensor) – (B, N, 5) Second box.
- 返回
(B, N) IoU.
- 返回类型
Tensor