mmcv.ops.diff_iou_rotated_2d

mmcv.ops.diff_iou_rotated_2d(box1: Tensor, box2: Tensor) → Tensor

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