Skip to content

GitLab

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

exporter changes

parent 26ba72bb
Show whitespace changes
Inline Side-by-side
Showing with 17 additions and 17 deletions
research/object_detection/exporter_lib_v2.py
View file @ f9ff935a
...@@ -62,7 +62,7 @@ class DetectionInferenceModule(tf.Module): ...@@ -62,7 +62,7 @@ class DetectionInferenceModule(tf.Module):
def __init__(self, detection_model, def __init__(self, detection_model,
use_side_inputs=False, use_side_inputs=False,
zipped_side_inputs=None: zipped_side_inputs=None):
"""Initializes a module for detection. """Initializes a module for detection.
Args: Args:
...@@ -116,7 +116,7 @@ class DetectionFromImageModule(DetectionInferenceModule): ...@@ -116,7 +116,7 @@ class DetectionFromImageModule(DetectionInferenceModule):
name=info[2])) name=info[2]))
def __call__(input_tensor, *side_inputs): 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) return self._run_inference_on_images(input_tensor, **kwargs)
self.__call__ = tf.function(__call__, input_signature=sig) 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