昇腾社区首页
中文
注册
开发者
下载

(beta)torch_npu.contrib.function.npu_diou

须知:[object Object] 该接口计划废弃,可以使用torch_npu.npu_diou接口进行替换。

产品支持情况

产品 是否支持
[object Object]Atlas A3 训练系列产品[object Object]
[object Object]Atlas A2 训练系列产品[object Object]
[object Object]Atlas 训练系列产品[object Object]
[object Object]Atlas 推理系列产品[object Object]

功能说明

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

函数原型

[object Object]

参数说明

  • boxes1Tensor):格式为xywh、shape为(4, n)的预测检测框。
  • boxes2Tensor):相应的gt检测框,shape为(4, n)。
  • transbool):是否有偏移,默认值为True。
  • is_crossbool):box1和box2之间是否有交叉操作,默认值为False。
  • modeint):选择DIoU的计算方式,默认值为0。

返回值说明

Tensor

IoU,size为[1,n]。

约束说明

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

调用示例

[object Object]