jaccard_index
jaccard_index(pred_poly: np.ndarray, gt_poly: np.ndarray, image_size: Tuple[int, int]) -> float
-
說明:
jaccard_index
是一個用來計算兩個多邊形之間的 Jaccard index 的函數。這個函數會計算兩個多邊形之間的交集面積與聯集面積之比。輸入的多邊形類型必須為np.ndarray
。 -
參數
- pred_poly (
np.ndarray
):預測的多邊形,一個 4 個點的多邊形。 - gt_poly (
np.ndarray
):真實的多邊形,一個 4 個點的多邊形。 - image_size (
Tuple[int, int]
):影像大小,(高度, 寬度)。
- pred_poly (
-
範例
import docsaidkit as D
pred_poly = np.array([[0, 0], [0, 1], [1, 1], [1, 0]])
gt_poly = np.array([[0.5, 0.5], [0.5, 1.5], [1.5, 1.5], [1.5, 0.5]])
image_size = (2, 2)
jaccard_index = D.jaccard_index(pred_poly, gt_poly, image_size)
print(jaccard_index)
# >>> 0.14285714285714285