torch_npu.npu_rotated_overlaps

API接口

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)