diff --git a/ssod/models/utils/bbox_utils.py b/ssod/models/utils/bbox_utils.py index 6614679..a811d80 100644 --- a/ssod/models/utils/bbox_utils.py +++ b/ssod/models/utils/bbox_utils.py @@ -236,7 +236,7 @@ def transform_image(img, M, out_shape): def filter_invalid(bbox, label=None, score=None, mask=None, thr=0.0, min_size=0): - if (score is not None) and (thr > 0): + if score is not None: valid = score > thr bbox = bbox[valid] if label is not None: