Skip to content

GitLab

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):
side_input_2 = np.ones((2, 2), dtype=np.uint8)
if (use_default_serving):
detections = detect_fn_sig(input_tensor=image,
side_inp_1=tf.constant(side_input_1),
side_inp_2=tf.constant(side_input_2))
side_inp_1=tf.constant(side_input_1),
side_inp_2=tf.constant(side_input_2))
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
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