Fix template names - coco convert

Author: dshabin

superannotate/input_converters/converters/coco_converters/coco_to_sa_vector.py

@@ -171,7 +171,8 @@ def coco_keypoint_detection_to_sa_vector(coco_path, output_dir):
         cat_id_to_cat[cat["id"]] = {
             "name": cat["name"],
             "keypoints": cat["keypoints"],
-            "skeleton": cat["skeleton"]
+            "skeleton": cat["skeleton"],
+            "supercategory": cat["supercategory"]
         }
 
     image_id_to_annotations = {}
@@ -239,8 +240,12 @@ def coco_keypoint_detection_to_sa_vector(coco_path, output_dir):
                     pointLabels[id_mapping[kp_index + 1] - 1] = kp_name
 
                 sa_obj = _create_vector_instance(
-                    'template', points, pointLabels, [],
-                    cat_id_to_cat[annot["category_id"]]["name"], connections
+                    'template',
+                    points,
+                    pointLabels, [],
+                    cat_id_to_cat[annot["category_id"]]["supercategory"],
+                    connections,
+                    template_name=cat_id_to_cat[annot["category_id"]]["name"]
                 )
                 if str(annot['image_id']) not in image_id_to_annotations:
                     image_id_to_annotations[str(annot['image_id'])] = [sa_obj]

superannotate/input_converters/converters/sa_json_helper.py

@@ -8,7 +8,8 @@ def _create_vector_instance(
     pointLabels,
     attributes,
     class_name='',
-    connections=[]
+    connections=[],
+    template_name=''
 ):
     sa_instance = {
         'type': instance_type,
@@ -20,6 +21,8 @@ def _create_vector_instance(
     if instance_type == 'template':
         sa_instance['points'] = points
         sa_instance['connections'] = connections
+        sa_instance['className'] = class_name
+        sa_instance['templateName'] = template_name
     elif instance_type == 'point':
         sa_instance['x'] = points[0]
         sa_instance['y'] = points[1]