"README_ORIGIN.md" did not exist on "11fb22aae80a60fd867ce87ff8795256b0f733cd"
Unverified Commit 420a7253 authored by pkulzc's avatar pkulzc Committed by GitHub
Browse files

Refactor tests for Object Detection API. (#8688)

Internal changes

--

PiperOrigin-RevId: 316837667
parent d0ef3913
...@@ -18,9 +18,11 @@ from __future__ import absolute_import ...@@ -18,9 +18,11 @@ from __future__ import absolute_import
from __future__ import division from __future__ import division
from __future__ import print_function from __future__ import print_function
import unittest
import tensorflow.compat.v1 as tf import tensorflow.compat.v1 as tf
from object_detection.core import standard_fields from object_detection.core import standard_fields
from object_detection.metrics import calibration_evaluation from object_detection.metrics import calibration_evaluation
from object_detection.utils import tf_version
def _get_categories_list(): def _get_categories_list():
...@@ -36,6 +38,7 @@ def _get_categories_list(): ...@@ -36,6 +38,7 @@ def _get_categories_list():
}] }]
@unittest.skipIf(tf_version.is_tf2(), 'Skipping TF1.X only test.')
class CalibrationDetectionEvaluationTest(tf.test.TestCase): class CalibrationDetectionEvaluationTest(tf.test.TestCase):
def _get_ece(self, ece_op, update_op): def _get_ece(self, ece_op, update_op):
......
This diff is collapsed.
This diff is collapsed.
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment