remove_small_boxes

torchvision.ops.remove_small_boxes(boxes: torch.Tensor, min_size: float) → torch.Tensor

Remove boxes which contains at least one side smaller than min_size.

Parameters

• boxes (Tensor[N, 4]) – boxes in (x1, y1, x2, y2) format with 0 <= x1 < x2 and 0 <= y1 < y2.

• min_size (float) – minimum size

Returns

indices of the boxes that have both sides larger than min_size

Return type

Tensor[K]