Merge branch 'json_enhacement' of https://github.com/superannotateai/superannotate-python-sdk into json_enhacement

Author: agyurjin

.pylintrc

@@ -1,5 +1,5 @@
 [BASIC]
-good-names=df,f,e,i,j,k
+good-names=df,f,e,i,j,k,pt
 
 bad-functions=
   apply,

superannotate/__init__.py

@@ -77,6 +77,7 @@
     coco_split_dataset, convert_project_type, export_annotation,
     import_annotation
 )
+from .old_to_new_format_convertor import update_json_format
 from .version import __version__
 
 formatter = logging.Formatter(fmt='SA-PYTHON-SDK - %(levelname)s - %(message)s')

superannotate/analytics/common.py

@@ -1,10 +1,12 @@
+import glob
 import json
 import logging
 from pathlib import Path
-import glob
 
 import pandas as pd
+
 from ..exceptions import SABaseException
+
 logger = logging.getLogger("superannotate-python-sdk")
 
 
@@ -29,7 +31,12 @@ def df_to_annotations(df, output_dir):
         image_height = None
         image_width = None
         image_df = df[df["imageName"] == image]
-        image_annotation = []
+        image_annotation = {
+            "instances": [],
+            "metadata": {},
+            "tags": [],
+            "comments": []
+        }
         instances = image_df["instanceId"].dropna().unique()
         for instance in instances:
             instance_df = image_df[image_df["instanceId"] == instance]
@@ -64,27 +71,31 @@ def df_to_annotations(df, output_dir):
                             "name": row["attributeName"]
                         }
                     )
-            image_annotation.append(instance_annotation)
+            image_annotation["instances"].append(instance_annotation)
             image_width = image_width or instance_df.iloc[0]["imageWidth"]
             image_height = image_height or instance_df.iloc[0]["imageHeight"]
             image_pinned = image_pinned or instance_df.iloc[0]["imagePinned"]
             image_status = image_status or instance_df.iloc[0]["imageStatus"]
 
         comments = image_df[image_df["type"] == "comment"]
         for _, comment in comments.iterrows():
-            comment_json = {"type": "comment"}
+            comment_json = {}
             comment_json.update(comment["meta"])
+            comment_json["correspondence"] = comment_json["comments"]
+            del comment_json["comments"]
             comment_json["resolved"] = comment["commentResolved"]
-            image_annotation.append(comment_json)
+            image_annotation["comments"].append(comment_json)
+
+        tags = image_df[image_df["type"] == "tag"]
+        for _, tag in tags.iterrows():
+            image_annotation["tags"].append(tag["tag"])
 
-        meta = {
-            "type": "meta",
+        image_annotation["metadata"] = {
             "width": int(image_width),
             "height": int(image_height),
             "status": image_status,
             "pinned": bool(image_pinned)
         }
-        image_annotation.append(meta)
         json.dump(
             image_annotation,
             open(output_dir / f"{image}___{project_suffix}", "w"),
@@ -224,13 +235,11 @@ def __append_annotation(annotation_dict):
 
     def __get_image_metadata(image_name, annotations):
         image_metadata = {"imageName": image_name}
-        for annotation in annotations:
-            if "type" in annotation and annotation["type"] == "meta":
-                image_metadata["imageHeight"] = annotation.get("height")
-                image_metadata["imageWidth"] = annotation.get("width")
-                image_metadata["imageStatus"] = annotation.get("status")
-                image_metadata["imagePinned"] = annotation.get("pinned")
-                break
+
+        image_metadata["imageHeight"] = annotations["metadata"].get("height")
+        image_metadata["imageWidth"] = annotations["metadata"].get("width")
+        image_metadata["imageStatus"] = annotations["metadata"].get("status")
+        image_metadata["imagePinned"] = annotations["metadata"].get("pinned")
         return image_metadata
 
     def __get_user_metadata(annotation):
@@ -277,38 +286,30 @@ def __get_user_metadata(annotation):
         image_name = annotation_path.name.split(type_postfix)[0]
         image_metadata = __get_image_metadata(image_name, annotation_json)
         annotation_instance_id = 0
-        for annotation in annotation_json:
+        if include_comments:
+            for annotation in annotation_json["comments"]:
+                comment_resolved = annotation["resolved"]
+                comment_meta = {
+                    "x": annotation["x"],
+                    "y": annotation["y"],
+                    "comments": annotation["correspondence"]
+                }
+                annotation_dict = {
+                    "type": "comment",
+                    "meta": comment_meta,
+                    "commentResolved": comment_resolved,
+                }
+                user_metadata = __get_user_metadata(annotation)
+                annotation_dict.update(user_metadata)
+                annotation_dict.update(image_metadata)
+                __append_annotation(annotation_dict)
+        if include_tags:
+            for annotation in annotation_json["tags"]:
+                annotation_dict = {"type": "tag", "tag": annotation}
+                annotation_dict.update(image_metadata)
+                __append_annotation(annotation_dict)
+        for annotation in annotation_json["instances"]:
             annotation_type = annotation.get("type", "mask")
-            if annotation_type == "meta":
-                continue
-            if annotation_type == "comment":
-                if include_comments:
-                    comment_resolved = annotation["resolved"]
-                    comment_meta = {
-                        "x": annotation["x"],
-                        "y": annotation["y"],
-                        "comments": annotation["comments"]
-                    }
-                    annotation_dict = {
-                        "type": annotation_type,
-                        "meta": comment_meta,
-                        "commentResolved": comment_resolved,
-                    }
-                    user_metadata = __get_user_metadata(annotation)
-                    annotation_dict.update(user_metadata)
-                    annotation_dict.update(image_metadata)
-                    __append_annotation(annotation_dict)
-                continue
-            if annotation_type == "tag":
-                if include_tags:
-                    annotation_tag = annotation["name"]
-                    annotation_dict = {
-                        "type": annotation_type,
-                        "tag": annotation_tag
-                    }
-                    annotation_dict.update(image_metadata)
-                    __append_annotation(annotation_dict)
-                continue
             annotation_class_name = annotation.get("className")
             if annotation_class_name is None or annotation_class_name not in class_name_to_color:
                 logger.warning(