昇腾社区首页
中文
注册

(beta)torch_npu.contrib.npu_giou

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

函数原型

[object Object]

功能说明

提供NPU版本的GIoU计算接口。

参数说明

  • 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]Atlas 训练系列产品[object Object]
  • [object Object]Atlas A2 训练系列产品[object Object]
  • [object Object]Atlas A3 训练系列产品[object Object]
  • [object Object]Atlas 推理系列产品[object Object]

调用示例

[object Object]