remove_small_boxes¶
- torchvision.ops.remove_small_boxes(boxes: Tensor, min_size: float) Tensor[原始碼]¶
移除
boxes中至少有一條邊長小於min_size的所有框。注意
對於清理
BoundingBoxes物件,請考慮改用變換SanitizeBoundingBoxes()。- 引數:
boxes (Tensor[N, 4]) – 採用
(x1, y1, x2, y2)格式的框,其中0 <= x1 < x2且0 <= y1 < y2。min_size (float) – 最小尺寸
- 返回:
兩側都大於
min_size的框的索引- 返回型別:
Tensor[K]