torch_npu.contrib.function.npu_bbox_coder_encode_yolo(bboxes, gt_bboxes, stride):
使用NPU OP获取将bbox转换为gt_bbox的框回归转换deltas。
torch.Tensor - 框转换deltas。
1 2 3 4 5 6 7 8 | >>> from torch_npu.contrib.function import npu_bbox_coder_encode_yolo >>> A = 1024 >>> bboxes = torch.randint(0, 512, size=(A, 4)).npu() >>> gt_bboxes = torch.randint(0, 512, size=(A, 4)).npu() >>> stride = torch.randint(0, 32, size=(A,)).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) |