def decode_fn(value): tensors = example_parser.decode_serialized_example( serialized_example=value, features=rio_specs.scene_feature_spec(with_annotations=True)) tensor_dict = tfds.core.utils.flatten_nest_dict(tensors) return tensor_dict
def decode_fn(value): tensors = example_parser.decode_serialized_example( serialized_example=value, features=waymo_frames.FRAME_FEATURE_SPEC) tensor_dict = tfds.core.utils.flatten_nest_dict(tensors) return tensor_dict