Skip to content

GitLab

"...git@developer.sourcefind.cn:OpenDAS/torchaudio.git" did not exist on "2d99fee29efac7fcd09f679ed6f7f3379eaac512"
Commit ad942e22 authored by Kaushik Shivakumar's avatar Kaushik Shivakumar
Browse files

fix line length

parent 811f3940
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 3 deletions
research/object_detection/exporter_lib_tf2_test.py
View file @ ad942e22
...@@ -237,10 +237,12 @@ class ExportInferenceGraphTest(tf.test.TestCase, parameterized.TestCase): ...@@ -237,10 +237,12 @@ class ExportInferenceGraphTest(tf.test.TestCase, parameterized.TestCase):
side_input_2 = np.ones((2, 2), dtype=np.uint8) side_input_2 = np.ones((2, 2), dtype=np.uint8)
if (use_default_serving): if (use_default_serving):
detections = detect_fn_sig(input_tensor=image, detections = detect_fn_sig(input_tensor=image,
side_inp_1=tf.constant(side_input_1), side_inp_1=tf.constant(side_input_1),
side_inp_2=tf.constant(side_input_2)) side_inp_2=tf.constant(side_input_2))
else: else:
detections = detect_fn(image, tf.constant(side_input_1), tf.constant(side_input_2)) detections = detect_fn(image,
tf.constant(side_input_1),
tf.constant(side_input_2))
detection_fields = fields.DetectionResultFields detection_fields = fields.DetectionResultFields
self.assertAllClose(detections[detection_fields.detection_boxes], self.assertAllClose(detections[detection_fields.detection_boxes],
......
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