remove_small_boxes¶
-
torchvision.ops.
remove_small_boxes
(boxes: torch.Tensor, min_size: float) → torch.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]