- torchvision.ops.remove_small_boxes(boxes: Tensor, min_size: float) Tensor [source]¶
Remove boxes which contains at least one side smaller than min_size.
boxes (Tensor[N, 4]) – boxes in
(x1, y1, x2, y2)format with
0 <= x1 < x2and
0 <= y1 < y2.
min_size (float) – minimum size
indices of the boxes that have both sides larger than min_size
- Return type: