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
7e4f1ef3
Unverified
Commit
7e4f1ef3
authored
Mar 06, 2022
by
srihari-humbarwadi
Browse files
rename features attr for `TfExampleDecoder`
parent
81edc4fe
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
2 deletions
+2
-2
official/vision/beta/projects/panoptic_maskrcnn/dataloaders/panoptic_deeplab_input.py
...s/panoptic_maskrcnn/dataloaders/panoptic_deeplab_input.py
+2
-2
No files found.
official/vision/beta/projects/panoptic_maskrcnn/dataloaders/panoptic_deeplab_input.py
View file @
7e4f1ef3
...
...
@@ -43,7 +43,7 @@ class TfExampleDecoder(tf_example_decoder.TfExampleDecoder):
include_mask
=
True
,
regenerate_source_id
=
regenerate_source_id
)
self
.
_
segmentation
_keys_to_features
=
{
self
.
_
panoptic
_keys_to_features
=
{
'image/panoptic/category_mask'
:
tf
.
io
.
FixedLenFeature
((),
tf
.
string
,
default_value
=
''
),
'image/panoptic/instance_mask'
:
...
...
@@ -54,7 +54,7 @@ class TfExampleDecoder(tf_example_decoder.TfExampleDecoder):
decoded_tensors
=
super
(
TfExampleDecoder
,
self
).
decode
(
serialized_example
)
parsed_tensors
=
tf
.
io
.
parse_single_example
(
serialized_example
,
self
.
_
segmentation
_keys_to_features
)
serialized_example
,
self
.
_
panoptic
_keys_to_features
)
category_mask
=
tf
.
io
.
decode_image
(
parsed_tensors
[
'image/panoptic/category_mask'
],
channels
=
1
)
...
...
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