"git@developer.sourcefind.cn:modelzoo/resnet50_tensorflow.git" did not exist on "9e9534e862876e615161d6ffe0e8f2752e1e3bca"
Commit a5621d27 authored by Kaushik Shivakumar's avatar Kaushik Shivakumar
Browse files

fix files

parent 4e94b00e
...@@ -133,11 +133,7 @@ DETECTION_MODULE_MAP = { ...@@ -133,11 +133,7 @@ DETECTION_MODULE_MAP = {
def export_inference_graph(input_type, def export_inference_graph(input_type,
pipeline_config, pipeline_config,
trained_checkpoint_dir, trained_checkpoint_dir,
output_directory, output_directory):
use_side_inputs,
side_input_shapes,
side_input_types,
side_input_names):
"""Exports inference graph for the model specified in the pipeline config. """Exports inference graph for the model specified in the pipeline config.
This function creates `output_directory` if it does not already exist, This function creates `output_directory` if it does not already exist,
......
...@@ -106,27 +106,6 @@ flags.DEFINE_string('output_directory', None, 'Path to write outputs.') ...@@ -106,27 +106,6 @@ flags.DEFINE_string('output_directory', None, 'Path to write outputs.')
flags.DEFINE_string('config_override', '', flags.DEFINE_string('config_override', '',
'pipeline_pb2.TrainEvalPipelineConfig ' 'pipeline_pb2.TrainEvalPipelineConfig '
'text proto to override pipeline_config_path.') 'text proto to override pipeline_config_path.')
flags.DEFINE_boolean('use_side_inputs', False,
'If True, uses side inputs as well as image inputs.')
flags.DEFINE_string('side_input_shapes', None,
'If use_side_inputs is True, this explicitly sets '
'the shape of the side input tensors to a fixed size. The '
'dimensions are to be provided as a comma-separated list '
'of integers. A value of -1 can be used for unknown '
'dimensions. A `/` denotes a break, starting the shape of '
'the next side input tensor. This flag is required if '
'using side inputs.')
flags.DEFINE_string('side_input_types', None,
'If use_side_inputs is True, this explicitly sets '
'the type of the side input tensors. The '
'dimensions are to be provided as a comma-separated list '
'of types, each of `string`, `integer`, or `float`. '
'This flag is required if using side inputs.')
flags.DEFINE_string('side_input_names', None,
'If use_side_inputs is True, this explicitly sets '
'the names of the side input tensors required by the model '
'assuming the names will be a comma-separated list of '
'strings. This flag is required if using side inputs.')
flags.mark_flag_as_required('pipeline_config_path') flags.mark_flag_as_required('pipeline_config_path')
flags.mark_flag_as_required('trained_checkpoint_dir') flags.mark_flag_as_required('trained_checkpoint_dir')
...@@ -140,8 +119,7 @@ def main(_): ...@@ -140,8 +119,7 @@ def main(_):
text_format.Merge(FLAGS.config_override, pipeline_config) text_format.Merge(FLAGS.config_override, pipeline_config)
exporter_lib_v2.export_inference_graph( exporter_lib_v2.export_inference_graph(
FLAGS.input_type, pipeline_config, FLAGS.trained_checkpoint_dir, FLAGS.input_type, pipeline_config, FLAGS.trained_checkpoint_dir,
FLAGS.output_directory, FLAGS.use_side_inputs, FLAGS.side_input_shapes, FLAGS.output_directory)
FLAGS.side_input_types, FLAGS.side_input_names)
if __name__ == '__main__': if __name__ == '__main__':
......
"""Setup script for object_detection."""
from setuptools import find_packages
from setuptools import setup
REQUIRED_PACKAGES = ['Pillow>=1.0', 'Matplotlib>=2.1', 'Cython>=0.28.1']
setup(
name='object_detection',
version='0.1',
install_requires=REQUIRED_PACKAGES,
include_package_data=True,
packages=[p for p in find_packages() if p.startswith('object_detection')],
description='Tensorflow Object Detection Library',
)
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