文档
注册

beta)torch_npu.npu_diou

接口原型

torch_npu.npu_diou(Tensor self, Tensor gtboxes, bool trans=False, bool is_cross=False, int mode=0) -> Tensor

功能描述

应用基于NPU的DIoU操作。考虑到目标之间距离,以及距离和范围的重叠率,不同目标或边界需趋于稳定。

参数说明

  • boxes1 (Tensor) - 格式为xywh、shape为(4, n)的预测检测框。
  • boxes2 (Tensor) - 相应的gt检测框,shape为(4, n)。
  • trans (Bool,默认值为False) - 是否有偏移。
  • is_cross (Bool,默认值为False) - box1和box2之间是否有交叉操作。
  • mode (Int,默认值为0) - 选择DIoU的计算方式。0表示IoU,1表示IoF。

输出说明

torch.Tensor (Tensor) - mask操作的结果。

约束说明

到目前为止,DIoU向后只支持当前版本中的trans==True、is_cross==False、mode==0('iou')。如果需要反向传播,确保参数正确。

调用示例

    >>> box1 = torch.randn(4, 32).npu()
    >>> box1.requires_grad = True
    >>> box2 = torch.randn(4, 32).npu()
    >>> box2.requires_grad = True
    >>> diou = torch_npu.contrib.function.npu_diou(box1, box2) 
    >>> l = diou.sum()
    >>> l.backward()
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词