(beta)torch_npu.contrib.npu_giou
须知:[object Object] 该接口计划废弃,可以使用
torch_npu.npu_giou接口进行替换。
产品支持情况
| 产品 | 是否支持 |
|---|---|
| [object Object]Atlas A3 训练系列产品[object Object] | √ |
| [object Object]Atlas A2 训练系列产品[object Object] | √ |
| [object Object]Atlas 训练系列产品[object Object] | √ |
| [object Object]Atlas 推理系列产品[object Object] | √ |
功能说明
提供NPU版本的GIoU计算接口。
函数原型
[object Object]
参数说明
- boxes1(
Tensor):格式为xywh、shape为(4, n)的预测检测框。 - boxes2(
Tensor):相应的gt检测框,shape为(4, n)。 - is_permuted(
bool):坐标值是否已经标准化。默认为True。
约束说明
仅trans=True(仅支持xywh,不支持xyxy),is_cross=False(仅支持a.shape==b.shape的场景,不支持((n,4),(m,4))的场景)。
调用示例
[object Object]