remove_small_boxes

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

從 boxes 中移除所有至少有一條邊長度小於 min_size 的框。

注意

為了對 BoundingBoxes 物件進行消毒，請考慮改用 SanitizeBoundingBoxes() 變換。

引數:

• boxes (Tensor[..., 4]) – 格式為 (x1, y1, x2, y2) 的框，其中 0 <= x1 < x2 和 0 <= y1 < y2。

• min_size (float) – 最小尺寸

返回:

兩條邊都大於 min_size 的框的索引

返回型別:

Tensor[K]