Skip to content

GitLab

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

exporter

parent f9ff935a
Show whitespace changes
Inline Side-by-side
Showing with 6 additions and 6 deletions
research/object_detection/exporter_lib_v2.py
View file @ d73569c5
...@@ -50,12 +50,12 @@ def _zip_side_inputs(side_input_shapes="", ...@@ -50,12 +50,12 @@ def _zip_side_inputs(side_input_shapes="",
Returns: Returns:
a zipped list of side input tuples. a zipped list of side input tuples.
""" """
side_input_shapes = list(map(lambda x: eval('[' + x + ']'), if (side_input_shapes)
side_input_shapes.split("/"))) side_input_shapes = list(map(lambda x: eval('[' + x + ']'), side_input_shapes.split("/")))
side_input_types = list(map(eval, side_input_types.split(","))) side_input_types = map(eval, side_input_types.split(","))
return zip(side_input_shapes, print(list(side_input_types))
side_input_types, #side_input_types = list(map(eval, side_input_types.split(",")))
side_input_names.split(",")) return zip(side_input_shapes, side_input_types, side_input_names.split(","))
class DetectionInferenceModule(tf.Module): class DetectionInferenceModule(tf.Module):
"""Detection Inference Module.""" """Detection Inference Module."""
......
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