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