torch_npu.npu_rotated_overlaps

torch_npu.npu_rotated_overlaps(self, query_boxes, trans=False) -> Tensor

计算旋转框的重叠面积。

• 参数解释：
  • self (Tensor) -梯度增量数据，shape为(B, 5, N)数据类型为float32的3D张量。
  • query_boxes (Tensor) - 标注框，shape为(B, 5, K) 数据类型为float32的3D张量。
  • trans (Bool，默认值为False) - 值为True表示“xyxyt”，值为False表示“xywht”。
• 约束条件：

  无

• 示例：

  >>> a=np.random.uniform(0,1,(1,3,5)).astype(np.float16)
  >>> b=np.random.uniform(0,1,(1,2,5)).astype(np.float16)
  >>> box1=torch.from_numpy(a).to("npu")
  >>> box2=torch.from_numpy(a).to("npu")
  >>> output = torch_npu.npu_rotated_overlaps(box1, box2, trans=False)
  >>> output
  tensor([[[0.0000, 0.1562, 0.0000],
          [0.1562, 0.3713, 0.0611],
          [0.0000, 0.0611, 0.0000]]], device='npu:0', dtype=torch.float16)