Skip to content

GitLab

Unverified Commit 7e4f1ef3 authored by srihari-humbarwadi's avatar srihari-humbarwadi
Browse files

rename features attr for `TfExampleDecoder`

parent 81edc4fe
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/vision/beta/projects/panoptic_maskrcnn/dataloaders/panoptic_deeplab_input.py
View file @ 7e4f1ef3
...@@ -43,7 +43,7 @@ class TfExampleDecoder(tf_example_decoder.TfExampleDecoder): ...@@ -43,7 +43,7 @@ class TfExampleDecoder(tf_example_decoder.TfExampleDecoder):
include_mask=True, include_mask=True,
regenerate_source_id=regenerate_source_id) regenerate_source_id=regenerate_source_id)
self._segmentation_keys_to_features = { self._panoptic_keys_to_features = {
'image/panoptic/category_mask': 'image/panoptic/category_mask':
tf.io.FixedLenFeature((), tf.string, default_value=''), tf.io.FixedLenFeature((), tf.string, default_value=''),
'image/panoptic/instance_mask': 'image/panoptic/instance_mask':
...@@ -54,7 +54,7 @@ class TfExampleDecoder(tf_example_decoder.TfExampleDecoder): ...@@ -54,7 +54,7 @@ class TfExampleDecoder(tf_example_decoder.TfExampleDecoder):
decoded_tensors = super(TfExampleDecoder, decoded_tensors = super(TfExampleDecoder,
self).decode(serialized_example) self).decode(serialized_example)
parsed_tensors = tf.io.parse_single_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( category_mask = tf.io.decode_image(
parsed_tensors['image/panoptic/category_mask'], channels=1) parsed_tensors['image/panoptic/category_mask'], channels=1)
......
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