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
3ab72765
Commit
3ab72765
authored
Jun 27, 2017
by
Derek Chow
Browse files
snake_case Decode function
parent
58417e76
Changes
5
Hide whitespace changes
Inline
Side-by-side
Showing
5 changed files
with
13 additions
and
14 deletions
+13
-14
object_detection/builders/input_reader_builder.py
object_detection/builders/input_reader_builder.py
+1
-1
object_detection/core/data_decoder.py
object_detection/core/data_decoder.py
+1
-2
object_detection/data_decoders/tf_example_decoder.py
object_detection/data_decoders/tf_example_decoder.py
+1
-1
object_detection/data_decoders/tf_example_decoder_test.py
object_detection/data_decoders/tf_example_decoder_test.py
+9
-9
object_detection/exporter.py
object_detection/exporter.py
+1
-1
No files found.
object_detection/builders/input_reader_builder.py
View file @
3ab72765
...
...
@@ -60,6 +60,6 @@ def build(input_reader_config):
capacity
=
input_reader_config
.
queue_capacity
,
min_after_dequeue
=
input_reader_config
.
min_after_dequeue
)
return
tf_example_decoder
.
TfExampleDecoder
().
D
ecode
(
string_tensor
)
return
tf_example_decoder
.
TfExampleDecoder
().
d
ecode
(
string_tensor
)
raise
ValueError
(
'Unsupported input_reader_config.'
)
object_detection/core/data_decoder.py
View file @
3ab72765
...
...
@@ -26,9 +26,8 @@ class DataDecoder(object):
"""Interface for data decoders."""
__metaclass__
=
ABCMeta
# TODO: snake_case this method.
@
abstractmethod
def
D
ecode
(
self
,
data
):
def
d
ecode
(
self
,
data
):
"""Return a single image and associated labels.
Args:
...
...
object_detection/data_decoders/tf_example_decoder.py
View file @
3ab72765
...
...
@@ -82,7 +82,7 @@ class TfExampleDecoder(data_decoder.DataDecoder):
slim_example_decoder
.
Tensor
(
'image/segmentation/object/class'
)),
}
def
D
ecode
(
self
,
tf_example_string_tensor
):
def
d
ecode
(
self
,
tf_example_string_tensor
):
"""Decodes serialized tensorflow example and returns a tensor dictionary.
Args:
...
...
object_detection/data_decoders/tf_example_decoder_test.py
View file @
3ab72765
...
...
@@ -64,7 +64,7 @@ class TfExampleDecoderTest(tf.test.TestCase):
})).
SerializeToString
()
example_decoder
=
tf_example_decoder
.
TfExampleDecoder
()
tensor_dict
=
example_decoder
.
D
ecode
(
tf
.
convert_to_tensor
(
example
))
tensor_dict
=
example_decoder
.
d
ecode
(
tf
.
convert_to_tensor
(
example
))
self
.
assertAllEqual
((
tensor_dict
[
fields
.
InputDataFields
.
image
].
get_shape
().
as_list
()),
[
None
,
None
,
3
])
...
...
@@ -84,7 +84,7 @@ class TfExampleDecoderTest(tf.test.TestCase):
})).
SerializeToString
()
example_decoder
=
tf_example_decoder
.
TfExampleDecoder
()
tensor_dict
=
example_decoder
.
D
ecode
(
tf
.
convert_to_tensor
(
example
))
tensor_dict
=
example_decoder
.
d
ecode
(
tf
.
convert_to_tensor
(
example
))
with
self
.
test_session
()
as
sess
:
tensor_dict
=
sess
.
run
(
tensor_dict
)
...
...
@@ -103,7 +103,7 @@ class TfExampleDecoderTest(tf.test.TestCase):
})).
SerializeToString
()
example_decoder
=
tf_example_decoder
.
TfExampleDecoder
()
tensor_dict
=
example_decoder
.
D
ecode
(
tf
.
convert_to_tensor
(
example
))
tensor_dict
=
example_decoder
.
d
ecode
(
tf
.
convert_to_tensor
(
example
))
self
.
assertAllEqual
((
tensor_dict
[
fields
.
InputDataFields
.
image
].
get_shape
().
as_list
()),
[
None
,
None
,
3
])
...
...
@@ -130,7 +130,7 @@ class TfExampleDecoderTest(tf.test.TestCase):
})).
SerializeToString
()
example_decoder
=
tf_example_decoder
.
TfExampleDecoder
()
tensor_dict
=
example_decoder
.
D
ecode
(
tf
.
convert_to_tensor
(
example
))
tensor_dict
=
example_decoder
.
d
ecode
(
tf
.
convert_to_tensor
(
example
))
self
.
assertAllEqual
((
tensor_dict
[
fields
.
InputDataFields
.
groundtruth_boxes
].
get_shape
().
as_list
()),
[
None
,
4
])
...
...
@@ -153,7 +153,7 @@ class TfExampleDecoderTest(tf.test.TestCase):
})).
SerializeToString
()
example_decoder
=
tf_example_decoder
.
TfExampleDecoder
()
tensor_dict
=
example_decoder
.
D
ecode
(
tf
.
convert_to_tensor
(
example
))
tensor_dict
=
example_decoder
.
d
ecode
(
tf
.
convert_to_tensor
(
example
))
self
.
assertAllEqual
((
tensor_dict
[
fields
.
InputDataFields
.
groundtruth_classes
].
get_shape
().
as_list
()),
...
...
@@ -176,7 +176,7 @@ class TfExampleDecoderTest(tf.test.TestCase):
})).
SerializeToString
()
example_decoder
=
tf_example_decoder
.
TfExampleDecoder
()
tensor_dict
=
example_decoder
.
D
ecode
(
tf
.
convert_to_tensor
(
example
))
tensor_dict
=
example_decoder
.
d
ecode
(
tf
.
convert_to_tensor
(
example
))
self
.
assertAllEqual
((
tensor_dict
[
fields
.
InputDataFields
.
groundtruth_area
].
get_shape
().
as_list
()),
[
None
])
...
...
@@ -197,7 +197,7 @@ class TfExampleDecoderTest(tf.test.TestCase):
})).
SerializeToString
()
example_decoder
=
tf_example_decoder
.
TfExampleDecoder
()
tensor_dict
=
example_decoder
.
D
ecode
(
tf
.
convert_to_tensor
(
example
))
tensor_dict
=
example_decoder
.
d
ecode
(
tf
.
convert_to_tensor
(
example
))
self
.
assertAllEqual
((
tensor_dict
[
fields
.
InputDataFields
.
groundtruth_is_crowd
].
get_shape
().
as_list
()),
...
...
@@ -220,7 +220,7 @@ class TfExampleDecoderTest(tf.test.TestCase):
})).
SerializeToString
()
example_decoder
=
tf_example_decoder
.
TfExampleDecoder
()
tensor_dict
=
example_decoder
.
D
ecode
(
tf
.
convert_to_tensor
(
example
))
tensor_dict
=
example_decoder
.
d
ecode
(
tf
.
convert_to_tensor
(
example
))
self
.
assertAllEqual
((
tensor_dict
[
fields
.
InputDataFields
.
groundtruth_difficult
].
get_shape
().
as_list
()),
...
...
@@ -263,7 +263,7 @@ class TfExampleDecoderTest(tf.test.TestCase):
'image/segmentation/object/class'
:
self
.
_Int64Feature
(
instance_segmentation_classes
)})).
SerializeToString
()
example_decoder
=
tf_example_decoder
.
TfExampleDecoder
()
tensor_dict
=
example_decoder
.
D
ecode
(
tf
.
convert_to_tensor
(
example
))
tensor_dict
=
example_decoder
.
d
ecode
(
tf
.
convert_to_tensor
(
example
))
self
.
assertAllEqual
((
tensor_dict
[
fields
.
InputDataFields
.
groundtruth_instance_masks
].
...
...
object_detection/exporter.py
View file @
3ab72765
...
...
@@ -101,7 +101,7 @@ def freeze_graph_with_def_protos(
def
_tf_example_input_placeholder
():
tf_example_placeholder
=
tf
.
placeholder
(
tf
.
string
,
shape
=
[],
name
=
'tf_example'
)
tensor_dict
=
tf_example_decoder
.
TfExampleDecoder
().
D
ecode
(
tensor_dict
=
tf_example_decoder
.
TfExampleDecoder
().
d
ecode
(
tf_example_placeholder
)
image
=
tensor_dict
[
fields
.
InputDataFields
.
image
]
return
tf
.
expand_dims
(
image
,
axis
=
0
)
...
...
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