Update inputs_annotations.py (#457)

Clarifai · Dec 3, 2024 · df05d67 · df05d67
1 parent 639535b
commit df05d67
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/clarifai/datasets/export/inputs_annotations.py b/clarifai/datasets/export/inputs_annotations.py
@@ -165,6 +165,14 @@ def _save_video_to_archive(self, new_archive: zipfile.ZipFile, hosted_url: str,
   def _save_annotation_to_archive(self, new_archive: zipfile.ZipFile, annot_data: List[Dict],
                                   file_name: str) -> None:
     """Gets the annotation response bytestring (from requests) and append to zip file."""
+    # Fill zero values for missing bounding box keys
+    for annot in annot_data:
+      if annot.get('regionInfo') and annot['regionInfo'].get('boundingBox'):
+        bbox = annot['regionInfo']['boundingBox']
+        bbox.setdefault('topRow', 0)
+        bbox.setdefault('leftCol', 0)
+        bbox.setdefault('bottomRow', 0)
+        bbox.setdefault('rightCol', 0)
     # Serialize the dictionary to a JSON string
     json_str = json.dumps(annot_data)
     # Convert the JSON string to bytes