file metadata import

Author: ninthreezy

datatorch/api/scripts/import_coco.py

@@ -130,6 +130,12 @@ def simplify_segmentation(segmentation: List[List[float]], tolerance: float = 1)
     }
 """
 
+_SET_FILE_METADATA = """
+    mutation SetFileMetadata($fileId: ID!, $key: String!, $value: JSON!) {
+        setFileMetadata(fileId: $fileId, key: $key, value: $value)
+    }
+"""
+
 
 def import_coco(
     file_path: str,
@@ -141,6 +147,8 @@ def import_coco(
     simplify_tolerance: float = 0,
     ignore_annotations_with_ids: bool = True,
     api: ApiClient = None,
+    upload_file_metadata: bool = False,
+    upload_anno_metadata: bool = False,
 ):
     if not import_segmentation and not import_bbox:
         _LOGGER.warning("Nothing to import. Both segmentation and bbox are disabled.")
@@ -219,6 +227,16 @@ def import_coco(
                 )
                 continue
 
+        # Upload file metadata if they want to
+        if upload_file_metadata:
+            file_metadata = coco_image.get("metadata", {})
+            if len(file_metadata) > 0:
+                for key, value in file_metadata.items():
+                    api.execute(
+                        _SET_FILE_METADATA,
+                        params={"fileId": dt_file.id, "key": key, "value": value},
+                    )
+
         coco_annotation_ids = coco.getAnnIds(
             catIds=coco_category_ids, imgIds=coco_image["id"]
         )
@@ -311,6 +329,14 @@ def import_coco(
                             }
                         )
 
+            # Metadata stuff
+            # need to upload the create annotations api first
+            # metadata = anno.get("metadata", {})
+            # for metadata_key, metadata_value in metadata.items():
+            #     print(metadata)
+            #     print(metadata_key)
+            #     print(metadata_value)
+
             if created_segmentation or created_bbox:
                 new_annotations.append(annotation)
 

setup.py

@@ -33,7 +33,7 @@
 
 setup(
     name="datatorch",
-    version="0.4.8.3",
+    version="0.4.8.4",
     description="A CLI and library for interacting with DataTorch.",
     author="DataTorch",
     author_email="[email protected]",