fix tests

research/object_detection/core/region_similarity_calculator.py

@@ -47,6 +47,10 @@ class RegionSimilarityCalculator(six.with_metaclass(ABCMeta, object)):
       boxlist1: BoxList holding N boxes.
       boxlist2: BoxList holding M boxes.
       scope: Op scope name. Defaults to 'Compare' if None.
+      groundtruth_labels: a Tensor of shape [num_boxes, num_classes]
+        containing groundtruth labels.
+      predicted_labels: a Tensor of shape [num_boxes, num_classes]
+        containing predicted labels.
 
     Returns:
       a (float32) tensor of shape [N, M] with pairwise similarity score.

research/object_detection/core/region_similarity_calculator_test.py

@@ -95,18 +95,18 @@ class RegionSimilarityCalculatorTest(test_case.TestCase):
 
   def test_detr_similarity(self):
     def graph_fn():
-      corners1 = tf.constant([[4.0, 3.0, 7.0, 5.0], [5.0, 6.0, 10.0, 7.0]])
-      corners2 = tf.constant([[3.0, 4.0, 6.0, 8.0], [14.0, 14.0, 15.0, 15.0],
-                              [0.0, 0.0, 20.0, 20.0]])
-      groundtruth_labels = tf.constant([[]])
+      corners1 = tf.constant([[5.0, 7.0, 7.0, 9.0]])
+      corners2 = tf.constant([[5.0, 7.0, 7.0, 9.0], [5.0, 11.0, 7.0, 13.0]])
+      groundtruth_labels = tf.constant([[1.0, 0.0]])
+      predicted_labels = tf.constant([[0.0, 1000.0], [1000.0, 0.0]])
       boxes1 = box_list.BoxList(corners1)
       boxes2 = box_list.BoxList(corners2)
-      iou_similarity_calculator = region_similarity_calculator.IouSimilarity()
-      iou_similarity = iou_similarity_calculator.compare(boxes1, boxes2)
-      return iou_similarity
-    exp_output = [[2.0 / 16.0, 0, 6.0 / 400.0], [1.0 / 16.0, 0.0, 5.0 / 400.0]]
-    iou_output = self.execute(graph_fn, [])
-    self.assertAllClose(iou_output, exp_output)
+      detr_similarity_calculator = region_similarity_calculator.DETRSimiliarity()
+      detr_similarity = detr_similarity_calculator.compare(boxes1, boxes2, None, groundtruth_labels, predicted_labels)
+      return detr_similarity
+    exp_output = [[2.0, -2.0/3.0 + 1.0 - 20.0]]
+    sim_output = self.execute(graph_fn, [])
+    self.assertAllClose(sim_output, exp_output)
 
 if __name__ == '__main__':
-  tf.test.main()
+  tf.test.main()
\ No newline at end of file