計算Tensorflow中的邊界框重疊

Question

假設我們有兩個邊界框座標的張量：N，尺寸爲[n，4]和K，尺寸爲[k，4]。每個張量的每一行代表邊界框的x1，y1，x2和y2。

Tensorflow中是否有一種有效方法來生成[n，k]矩陣M其中M[i, j] = overlap(N[i, :], K[j, :])？理想情況下，overlap將是交匯點交匯點，但如果使用其他方法也可以更簡單。

Answer 1

官方物體檢測模型的source code具有計算IOU的函數，見功能intersection和iou參考