Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
ResNet50_tensorflow
Commits
a5621d27
Commit
a5621d27
authored
Jul 24, 2020
by
Kaushik Shivakumar
Browse files
fix files
parent
4e94b00e
Changes
4
Show whitespace changes
Inline
Side-by-side
Showing
4 changed files
with
18 additions
and
28 deletions
+18
-28
research/object_detection/dataset_tools/create_ava_actions_tf_record.py
...t_detection/dataset_tools/create_ava_actions_tf_record.py
+0
-0
research/object_detection/exporter_lib_v2.py
research/object_detection/exporter_lib_v2.py
+1
-5
research/object_detection/exporter_main_v2.py
research/object_detection/exporter_main_v2.py
+1
-23
research/setup.py
research/setup.py
+16
-0
No files found.
research/object_detection/dataset_tools/create_ava_tf_record.py
→
research/object_detection/dataset_tools/create_ava_
actions_
tf_record.py
View file @
a5621d27
File moved
research/object_detection/exporter_lib_v2.py
View file @
a5621d27
...
...
@@ -133,11 +133,7 @@ DETECTION_MODULE_MAP = {
def
export_inference_graph
(
input_type
,
pipeline_config
,
trained_checkpoint_dir
,
output_directory
,
use_side_inputs
,
side_input_shapes
,
side_input_types
,
side_input_names
):
output_directory
):
"""Exports inference graph for the model specified in the pipeline config.
This function creates `output_directory` if it does not already exist,
...
...
research/object_detection/exporter_main_v2.py
View file @
a5621d27
...
...
@@ -106,27 +106,6 @@ flags.DEFINE_string('output_directory', None, 'Path to write outputs.')
flags
.
DEFINE_string
(
'config_override'
,
''
,
'pipeline_pb2.TrainEvalPipelineConfig '
'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
(
'trained_checkpoint_dir'
)
...
...
@@ -140,8 +119,7 @@ def main(_):
text_format
.
Merge
(
FLAGS
.
config_override
,
pipeline_config
)
exporter_lib_v2
.
export_inference_graph
(
FLAGS
.
input_type
,
pipeline_config
,
FLAGS
.
trained_checkpoint_dir
,
FLAGS
.
output_directory
,
FLAGS
.
use_side_inputs
,
FLAGS
.
side_input_shapes
,
FLAGS
.
side_input_types
,
FLAGS
.
side_input_names
)
FLAGS
.
output_directory
)
if
__name__
==
'__main__'
:
...
...
research/setup.py
0 → 100644
View file @
a5621d27
"""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'
,
)
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment