diff --git a/ppdet/modeling/bbox_utils.py b/ppdet/modeling/bbox_utils.py index aa55538f31fb5c04cc4e181195acc4ff2289cbd7..4c4acd9dd7a5ead687f4af33d1b6011c9d9bb43f 100644 --- a/ppdet/modeling/bbox_utils.py +++ b/ppdet/modeling/bbox_utils.py @@ -100,7 +100,7 @@ def clip_bbox(boxes, im_shape): def nonempty_bbox(boxes, min_size=0, return_mask=False): w = boxes[:, 2] - boxes[:, 0] h = boxes[:, 3] - boxes[:, 1] - mask = paddle.logical_and(w > min_size, w > min_size) + mask = paddle.logical_and(h > min_size, w > min_size) if return_mask: return mask keep = paddle.nonzero(mask).flatten()