def npu_bbox_coder_encode_yolo()

API接口

def npu_bbox_coder_encode_yolo(bboxes, gt_bboxes, stride):

功能描述

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

参数说明

输出说明

torch.Tensor - 框转换deltas。

示例

调用方式示例： 
from torch_npu.contrib.function import npu_bbox_coder_encode_yolo
out = npu_bbox_coder_encode_yolo(bboxes, gt_bboxes, stride)
使用示例： 
    >>> 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)
父主题： torch_npu.contrib