Ejemplo n.º 1
0
 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