Skip to content

Commit c636ea3

Browse files
Don't require image/source_id feature if regenerate_source_id is set.
PiperOrigin-RevId: 395539392
1 parent dc6c341 commit c636ea3

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

official/vision/beta/dataloaders/tf_example_decoder.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,6 @@ def __init__(self,
3838
self._regenerate_source_id = regenerate_source_id
3939
self._keys_to_features = {
4040
'image/encoded': tf.io.FixedLenFeature((), tf.string),
41-
'image/source_id': tf.io.FixedLenFeature((), tf.string),
4241
'image/height': tf.io.FixedLenFeature((), tf.int64),
4342
'image/width': tf.io.FixedLenFeature((), tf.int64),
4443
'image/object/bbox/xmin': tf.io.VarLenFeature(tf.float32),
@@ -54,6 +53,10 @@ def __init__(self,
5453
self._keys_to_features.update({
5554
'image/object/mask': tf.io.VarLenFeature(tf.string),
5655
})
56+
if not regenerate_source_id:
57+
self._keys_to_features.update({
58+
'image/source_id': tf.io.FixedLenFeature((), tf.string),
59+
})
5760

5861
def _decode_image(self, parsed_tensors):
5962
"""Decodes the image and set its static shape."""

0 commit comments

Comments
 (0)