[object Object]
应用基于NPU的DIoU操作。考虑到目标之间距离,以及距离和范围的重叠率,不同目标或边界需趋于稳定。
[object Object]
- boxes1(
[object Object]):格式为xywh、shape为(4, n)的预测检测框。 - boxes2(
[object Object]):相应的gt检测框,shape为(4, n)。 - trans(
[object Object]):是否有偏移,默认值为True。 - is_cross(
[object Object]):box1和box2之间是否有交叉操作,默认值为False。 - mode(
[object Object]):选择DIoU的计算方式,默认值为0。
[object Object]
IoU,size为[1,n]。
到目前为止,DIoU反向只支持当前版本中的trans==True、is_cross==False、mode==0('iou')。如果需要反向传播,确保参数正确。
[object Object]