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