快捷方式

box_iou

torchvision.ops.box_iou(boxes1: Tensor, boxes2: Tensor) Tensor[source]

返回兩組邊界框之間的交併比(Jaccard 指數)。

兩組邊界框都應採用 (x1, y1, x2, y2) 格式,並滿足 0 <= x1 < x20 <= y1 < y2 的條件。

引數:
  • boxes1 (Tensor[N, 4]) – 第一組邊界框

  • boxes2 (Tensor[M, 4]) – 第二組邊界框

返回值:

一個 NxM 矩陣,包含 boxes1 中每個元素與 boxes2 中每個元素之間的成對 IoU 值

返回型別:

Tensor[N, M]

文件

訪問 PyTorch 的完整開發者文件

檢視文件

教程

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

檢視教程

資源

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

檢視資源