Beispiel #1
0
def intersection(boxlist1, boxlist2):
    """Compute pairwise intersection areas between boxes.

  Args:
    boxlist1: BoxList holding N boxes
    boxlist2: BoxList holding M boxes

  Returns:
    a numpy array with shape [N*M] representing pairwise intersection area
  """
    return np_box_ops.intersection(boxlist1.get(), boxlist2.get())
Beispiel #2
0
 def testIntersection(self):
     intersection = np_box_ops.intersection(self.boxes1, self.boxes2)
     expected_intersection = np.array([[2.0, 0.0, 6.0], [1.0, 0.0, 5.0]],
                                      dtype=float)
     self.assertAllClose(intersection, expected_intersection)