distance_box_iou_loss

torchvision.ops.distance_box_iou_loss(boxes1: Tensor, boxes2: Tensor, reduction: str = 'none', eps: float = 1e-07) → Tensor

梯度友好的 IoU 损失，具有额外的惩罚项，当框中心之间的距离不为零时，该惩罚项非零。实际上，对于两个完全重叠的框，距离 IoU 与 IoU 损失相同。此损失是对称的，因此 boxes1 和 boxes2 参数可以互换。

两组框都应为 (x1, y1, x2, y2) 格式，其中 0 <= x1 < x2 和 0 <= y1 < y2，并且两个框应具有相同的尺寸。

参数:

• boxes1 (Tensor[N, 4]) – 第一组框

• boxes2 (Tensor[N, 4]) – 第二组框

• reduction (string, optional) – 指定应用于输出的规约： 'none' | 'mean' | 'sum'。 'none'：不对输出应用任何规约。 'mean'：输出将被平均。 'sum'：输出将被求和。 默认： 'none'

• eps (float, optional) – 防止除以零的小数。 默认： 1e-7

返回:

带有归约选项应用的损失张量。

返回类型:

张量

参考

Zhaohui Zheng 等人：Distance Intersection over Union Loss： https://arxiv.org/abs/1911.08287