CroissantBuilder: Get the bounding box format from the field's source.

PiperOrigin-RevId: 784167034

Author: The TensorFlow Datasets Authors

tensorflow_datasets/core/dataset_builders/croissant_builder.py

@@ -50,6 +50,7 @@
 from tensorflow_datasets.core import splits as splits_lib
 from tensorflow_datasets.core.features import audio_feature
 from tensorflow_datasets.core.features import bounding_boxes
+from tensorflow_datasets.core.features import bounding_boxes_utils as bb_utils
 from tensorflow_datasets.core.features import feature as feature_lib
 from tensorflow_datasets.core.features import features_dict
 from tensorflow_datasets.core.features import image_feature
@@ -175,8 +176,17 @@ def datatype_converter(
     feature = image_feature.Image(doc=field.description)
   elif field_data_type == mlc.DataType.BOUNDING_BOX:
     # TFDS uses REL_YXYX by default, but Hugging Face doesn't enforce a format.
+    if bbox_format := field.source.format:
+      try:
+        bbox_format = bb_utils.BBoxFormat(bbox_format)
+      except ValueError as e:
+        raise ValueError(
+            f'Unsupported bounding box format: {bbox_format}. Currently'
+            ' supported bounding box formats are: '
+            f'{[format.value for format in bb_utils.BBoxFormat]}'
+        ) from e
     feature = bounding_boxes.BBoxFeature(
-        doc=field.description, bbox_format=None
+        doc=field.description, bbox_format=bbox_format
     )
   elif field_data_type == mlc.DataType.AUDIO_OBJECT:
     feature = audio_feature.Audio(

tensorflow_datasets/core/dataset_builders/croissant_builder_test.py

@@ -22,6 +22,7 @@
 from tensorflow_datasets.core.dataset_builders import croissant_builder
 from tensorflow_datasets.core.features import audio_feature
 from tensorflow_datasets.core.features import bounding_boxes
+from tensorflow_datasets.core.features import bounding_boxes_utils as bb_utils
 from tensorflow_datasets.core.features import features_dict
 from tensorflow_datasets.core.features import image_feature
 from tensorflow_datasets.core.features import sequence_feature
@@ -130,6 +131,27 @@ def test_simple_datatype_converter(
   assert actual_feature == expected_feature
 
 
+def test_bbox_datatype_converter():
+  field = mlc.Field(
+      data_types=mlc.DataType.BOUNDING_BOX,
+      description="Bounding box feature",
+      source=mlc.Source(format="XYWH"),
+  )
+  actual_feature = croissant_builder.datatype_converter(field)
+  assert isinstance(actual_feature, bounding_boxes.BBoxFeature)
+  assert actual_feature.bbox_format == bb_utils.BBoxFormat.XYWH
+
+
+def test_bbox_datatype_converter_with_invalid_format():
+  field = mlc.Field(
+      data_types=mlc.DataType.BOUNDING_BOX,
+      description="Bounding box feature",
+      source=mlc.Source(format="InvalidFormat"),
+  )
+  with pytest.raises(ValueError, match="Unsupported bounding box format"):
+    croissant_builder.datatype_converter(field)
+
+
 @pytest.mark.parametrize(
     ["field", "feature_type", "subfield_types"],
     [