From 11db901f639394fdcfca868cb781dae45054d67b Mon Sep 17 00:00:00 2001 From: Vasilis Vryniotis Date: Mon, 28 Mar 2022 12:49:35 +0100 Subject: [PATCH] Remove assertions for generalized_box_iou --- torchvision/ops/boxes.py | 6 ------ 1 file changed, 6 deletions(-) diff --git a/torchvision/ops/boxes.py b/torchvision/ops/boxes.py index 2f1f984ca25..23c1001438c 100644 --- a/torchvision/ops/boxes.py +++ b/torchvision/ops/boxes.py @@ -298,12 +298,6 @@ def generalized_box_iou(boxes1: Tensor, boxes2: Tensor) -> Tensor: """ if not torch.jit.is_scripting() and not torch.jit.is_tracing(): _log_api_usage_once(generalized_box_iou) - # degenerate boxes gives inf / nan results - # so do an early check - if (boxes1[:, 2:] < boxes1[:, :2]).any(): - raise ValueError("Some of the input boxes1 are invalid.") - if not (boxes2[:, 2:] >= boxes2[:, :2]).all(): - raise ValueError("Some of the input boxes2 are invalid.") inter, union = _box_inter_union(boxes1, boxes2) iou = inter / union