remove_small_boxes

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

Remove every box from boxes which contains at least one side length that is smaller than min_size.

Note

For sanitizing a BoundingBoxes object, consider using the transform SanitizeBoundingBoxes() instead.

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]