Shortcuts

mmcv.ops.diff_iou_rotated_3d

mmcv.ops.diff_iou_rotated_3d(box3d1: torch.Tensor, box3d2: torch.Tensor)torch.Tensor[source]

Calculate differentiable iou of rotated 3d boxes.

Parameters
  • box3d1 (Tensor) – (B, N, 3+3+1) First box (x,y,z,w,h,l,alpha).

  • box3d2 (Tensor) – (B, N, 3+3+1) Second box (x,y,z,w,h,l,alpha).

Returns

(B, N) IoU.

Return type

Tensor