From 1b63c0a428d33288de56c454c9bce8cf04a76a72 Mon Sep 17 00:00:00 2001 From: Yonghye Kwon Date: Fri, 12 Feb 2021 17:19:45 +0900 Subject: [PATCH] check inf value for computing iou check inf value for computing iou --- torchvision/ops/boxes.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/torchvision/ops/boxes.py b/torchvision/ops/boxes.py index 1a9657f41ab..9f90b83b7ff 100644 --- a/torchvision/ops/boxes.py +++ b/torchvision/ops/boxes.py @@ -198,7 +198,10 @@ def _box_inter_union(boxes1: Tensor, boxes2: Tensor) -> Tuple[Tensor, Tensor]: inter = wh[:, :, 0] * wh[:, :, 1] # [N,M] union = area1[:, None] + area2 - inter - + + assert torch.count_nonzero(torch.isinf(inter)) == 0 + assert torch.count_nonzero(torch.isinf(union)) == 0 + return inter, union