nms¶
- torchvision.ops.nms(boxes: Tensor, scores: Tensor, iou_threshold: float) Tensor[原始碼]¶
根據方塊的交集重疊率 (IoU) 在方塊上執行非極大值抑制 (NMS)。
NMS 會迭代地移除 IoU 大於
iou_threshold的另一個（得分較高的）方塊的得分較低的方塊。如果多個方塊具有完全相同的分數，並且滿足關於參考方塊的 IoU 準則，則無法保證 CPU 和 GPU 之間選擇的方塊相同。這類似於 PyTorch 中存在重複值時 argsort 的行為。
- 參數:
boxes (Tensor[N, 4])) – 要在其上執行 NMS 的方塊。它們應採用
(x1, y1, x2, y2)格式，且0 <= x1 < x2和0 <= y1 < y2。scores (Tensor[N]) – 每個方塊的分數
iou_threshold (float) – 捨棄所有 IoU > iou_threshold 的重疊方塊
- 傳回:
int64 張量，包含 NMS 保留的元素的索引，並按分數降序排序
- 傳回類型:
Tensor
