IouV2

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	√
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√
Atlas 200I/500 A2 推理产品	×
Atlas 推理系列产品	√
Atlas 训练系列产品	×
Kirin X90 处理器系列产品	√
Kirin 9030 处理器系列产品	√

功能说明

• 算子功能：计算两个矩阵的重叠面积占两个矩阵总面积的比例，设预测框的左上角坐标为（X1，Y1），右下角坐标为（X2，Y2），真实框的左上角坐标为（X3，Y3），右下角坐标为（X4，Y4）。

• 计算公式：

  $$\begin{gathered} IOU=\frac{Are{a}_3}{Are{a}_1+Are{a}_2-Are{a}_3} \\ IOF=\frac{Are{a}_3}{Are{a}_2} \end{gathered}$$

  其中，Area_1为bBox的面积，Area_2为gtBox的面积，Area_3为两者重叠部分面积，x和y的定义见参数说明。

  $$\begin{gathered} Are{a}_1=(X_2-X_1)(Y_2-Y_1) \\ Are{a}_2=(X_4-X_3)(Y_4-Y_3) \\ Are{a}_3=max(min(X_2,X_4)-max(X_1,X_3),0)\ast max(min(Y_2,Y_4)-max(Y_1,Y_3),0) \end{gathered}$$

  IOF定义为前景交叉比，计算公式：

  $$IOF=\frac{Are{a}_3}{前景面积}$$

  对于库上算子，前景面积为Area2；对于竞品mmcv，前景面积为Area1。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
bboxes	输入	预测矩形框。	FLOAT32、FLOAT16、BFLOAT16	ND
gtboxes	输入	真值矩形框。	FLOAT32、FLOAT16、BFLOAT16	ND
eps	属性	防止除零，计算面积时，长和宽都会加上eps。	FLOAT32	-
aligned	属性	用于标识两个输入的shape是否相同。	BOOL	-
mode	属性	用于选择计算方式"iou"或"iof"。	String	-
overlap	输出	根据两个输入计算得到的交并比/前景交叉比。	FLOAT32、FLOAT16、BFLOAT16	ND

• Kirin X90/Kirin 9030 处理器系列产品: 不支持BFLOAT16。

约束说明

• 输入shape限制：输入shape为(N, 4)的二维张量，第二维的四个值(X1, Y1, X2, Y2)需满足X1 < X2, Y1 < Y2

调用说明

调用方式	样例代码	说明
aclnn接口	test_aclnn_iou_v2	通过aclnnIou接口方式调用IouV2算子。