def npu_bbox_coder_encode_yolo()

def npu_bbox_coder_encode_yolo(bboxes, gt_bboxes, stride):

使用NPU OP获取将bbox转换为gt_bbox的框回归转换deltas。

• 参数:
  • bboxes (torch.Tensor) - 源框，例如锚点框。支持数据类型：float、half。
  • gt_bboxes (torch.Tensor) - 转换目标框，例如真值框。支持数据类型：float、half。
  • stride (torch.Tensor) - Bbox步长。仅支持int张量。
• 调用方式样例：

  from torch_npu.contrib.function import npu_bbox_coder_encode_yolo
  out = npu_bbox_coder_encode_yolo(bboxes, gt_bboxes, stride)

• 返回值:

  torch.Tensor - 框转换deltas。

• 示例：

      >>> A = 1024
      >>> bboxes = torch.randint(0, 512, size=(A, 4))
      >>> gt_bboxes = torch.randint(0, 512, size=(A, 4))
      >>> stride = torch.randint(0, 32, size=(A,))
      >>> torch.npu.set_device(0)
      >>> bboxes = bboxes.npu()
      >>> gt_bboxes = gt_bboxes.npu()
      >>> stride = stride.npu()
      >>> out = npu_bbox_coder_encode_yolo(bboxes, gt_bboxes, stride)
      >>> torch.npu.synchronize()
      >>> print('_npu_bbox_coder_encode_yolo done. output shape is ', out.shape)