(beta)torch_npu.contrib.npu_iou
须知:[object Object] 该接口计划废弃,可以使用
torch_npu.npu_iou接口进行替换。
产品支持情况
| 产品 | 是否支持 |
|---|---|
| [object Object]Atlas A3 训练系列产品[object Object] | √ |
| [object Object]Atlas A2 训练系列产品[object Object] | √ |
| [object Object]Atlas 训练系列产品[object Object] | √ |
| [object Object]Atlas 推理系列产品[object Object] | √ |
功能说明
提供NPU版本的IoU计算操作。计算时会为重叠区域添加极小值,避免除零问题。
函数原型
[object Object]
参数说明
- boxes1(
Tensor):shape为(n, 4)的预测检测框。 - boxes2(
Tensor):shape为(m, 4)的预测检测框。 - is_normalized(
bool):坐标值是否已经标准化。默认为False。 - normalized_scale(
float):设置恢复坐标的标准化比例,默认100。
约束说明
该函数常用于bbox和anchor匹配时。到目前为止,这个函数还没有对应的后向运算符,所以不能用在IOU_Loss中,由于计算公式中分母加上了0.001以避免除以0,当输入框是归一化数据时,0.001的占比太大。此时需要放大输入值,避免0.001影响过大。
调用示例
[object Object]