Berechnen von Begrenzungsrahmenüberlappungen in Tensorflow

Question

Nehmen wir an, wir haben zwei Tensoren für Bounding Box-Koordinaten: N, mit Dimension [n, 4] und K, mit Dimension [k, 4]. Jede Zeile jedes Tensors repräsentiert die x1, y1, x2 und y2 einer Bounding Box.

Gibt es eine effiziente Methode in Tensorflow, um eine [n, k] -Matrix zu erzeugen M wo M[i, j] = overlap(N[i, :], K[j, :])? Im Idealfall wäre overlap die Schnittmenge über Union, aber wenn es einfacher ist, eine andere Methode zu verwenden, die auch funktionieren würde.

Answer 1

die source code offiziellen Objekterfassungsmodell hat eine Funktion zur Berechnung der IOU siehe Funktion intersection und iou Referenz