remove_small_boxes

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

从boxes中移除每个包含至少一个边长小于min_size的盒子。

注意

为了清理一个BoundingBoxes对象，考虑使用转换SanitizeBoundingBoxes()代替。

Parameters:

• boxes (Tensor[N, 4]) – 以(x1, y1, x2, y2)格式表示的框 其中0 <= x1 < x2 且 0 <= y1 < y2。

• min_size (float) – 最小尺寸

Returns:

具有两边都大于min_size的框的索引

Return type:

张量[K]