updates

2020-03-13 11:03:39 -07:00
parent 6aae5aca64
commit 0de07da612
2 changed files with 2 additions and 2 deletions
@@ -325,7 +325,7 @@ def box_iou(boxes1, boxes2):
    lt = torch.max(boxes1[:, None, :2], boxes2[:, :2])  # [N,M,2]
    rb = torch.min(boxes1[:, None, 2:], boxes2[:, 2:])  # [N,M,2]

-    inter = (rb - lt).clamp(min=0).prod(2)  # [N,M]
+    inter = (rb - lt).clamp(0).prod(2)  # [N,M]
    return inter / (area1[:, None] + area2 - inter)  # iou = inter / (area1 + area2 - inter)