跳至主要内容

pairwise_ioa

pairwise_ioa(boxes1: Boxes, boxes2: Boxes) -> np.ndarray

  • 說明

    pairwise_ioa 是一個用來計算兩個邊界框列表之間的 IoA (交集比例) 的函數。這個函數會計算所有 N x M 對的邊界框之間的 IoA。輸入的邊界框類型必須為 Boxes

  • 參數

    • boxes1 (Boxes):第一個邊界框列表。包含 N 個邊界框。
    • boxes2 (Boxes):第二個邊界框列表。包含 M 個邊界框。
  • 範例

    import docsaidkit as D

    boxes1 = D.Boxes([[10, 20, 50, 80], [20, 30, 60, 90]])
    boxes2 = D.Boxes([[20, 30, 60, 90], [30, 40, 70, 100]])
    ioa = D.pairwise_ioa(boxes1, boxes2)
    print(ioa)
    # >>> [[0.625 0.33333334]
    # [1.0 0.625]]

補充說明

IoA 簡介

IoA(Intersection over Area)是一種用來評估邊界框重疊程度的指標,它衡量的是預測邊界框和真實邊界框之間的交集面積與真實邊界框的面積之比。

定義

IoA 計算公式為預測邊界框和真實邊界框交集的面積除以真實邊界框的面積。 IoA 的值範圍從 0 到 1,數值越大表示預測邊界框覆蓋真實邊界框的程度越高。

計算步驟

  1. 確定邊界框座標:首先,需要確定預測邊界框和真實邊界框在影像中的位置,通常使用四個座標來表示一個邊界框:(x0, y0, x1, y1),其中 (x0, y0) 是邊界框左上角的座標,(x1, y1) 是右下角的座標。

  2. 計算交集面積:計算預測邊界框和真實邊界框的交集面積。

  3. 計算 IoA:交集面積除以真實邊界框的面積,得到 IoA 值。

應用場景

  • 目標偵測:在目標偵測任務中,IoA 用於評估預測邊界框和真實邊界框之間的重疊程度,進而評估偵測模型的準確性。

  • 模型評估:IoA 常用於評估和比較不同物件偵測模型的效能,較高的 IoA 值表示模型具有較好的偵測精度。

  • 非極大值抑制(NMS):在目標檢測後處理中,IoA 用於非極大值抑制,以消除重疊的檢測框,保留最佳的檢測結果。

優點與限制

  • 優點:IoA 能夠量化預測邊界框和真實邊界框之間的重疊程度,幫助評估模型的準確性。

  • 局限:IoA 只考慮了預測邊界框和真實邊界框之間的重疊程度,並未考慮其他因素,如邊界框的形狀和方向,因此在某些情況下可能不夠準確。