Commit f9ff935a authored by Kaushik Shivakumar's avatar Kaushik Shivakumar
Browse files

exporter changes

parent 26ba72bb
......@@ -62,7 +62,7 @@ class DetectionInferenceModule(tf.Module):
def __init__(self, detection_model,
use_side_inputs=False,
zipped_side_inputs=None:
zipped_side_inputs=None):
"""Initializes a module for detection.
Args:
......@@ -116,7 +116,7 @@ class DetectionFromImageModule(DetectionInferenceModule):
name=info[2]))
def __call__(input_tensor, *side_inputs):
kwargs = dict(zip(self.side_input_names.split(","), side_inputs))
kwargs = dict(zip(self.side_input_names, side_inputs))
return self._run_inference_on_images(input_tensor, **kwargs)
self.__call__ = tf.function(__call__, input_signature=sig)
......
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