快捷方式

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 < x20 <= y1 < y2

  • min_size (float) – 最小尺寸

返回:

兩側都大於 min_size 的框的索引

返回型別:

Tensor[K]

文件

訪問 PyTorch 的全面開發者文件

檢視文件

教程

獲取面向初學者和高階開發者的深度教程

檢視教程

資源

查詢開發資源並獲得問題解答

檢視資源