bug fixed

Author: agyurjin

superannotate/input_converters/converters/coco_converters/coco_to_sa_pixel.py

@@ -15,6 +15,10 @@ def coco_panoptic_segmentation_to_sa_pixel(coco_path, images_path):
     hex_colors = blue_color_generator(len(coco_json["categories"]))
     annotate_list = coco_json["annotations"]
 
+    cat_id_to_cat = {}
+    for cat in coco_json['categories']:
+        cat_id_to_cat[cat['id']] = cat['name']
+
     sa_jsons = {}
     for annotate in tqdm(annotate_list, "Converting"):
         annot_name = os.path.splitext(annotate["file_name"])[0]
@@ -39,6 +43,7 @@ def coco_panoptic_segmentation_to_sa_pixel(coco_path, images_path):
                        axis=1)] = hex_to_rgb(hex_colors[i])
             dd = {
                 "classId": seg["category_id"],
+                'className': cat_id_to_cat[seg["category_id"]],
                 "probability": 100,
                 "visible": True,
                 "parts": [{

superannotate/input_converters/converters/coco_converters/coco_to_sa_vector.py

@@ -230,13 +230,19 @@ def coco_keypoint_detection_to_sa_vector(coco_path, images_path):
                         }
                     )
 
-                for point in sa_points:
-                    point_index = sa_points.index(point)
+                for point_index, point in enumerate(sa_points):
+                    if sa_points[point_index] == (-17, -17):
+                        x = 5
+                        y = 5
+                    else:
+                        x = sa_points[point_index][0],
+                        y = sa_points[point_index][1]
+
                     sa_template['points'].append(
                         {
                             'id': point_index + 1,
-                            'x': sa_points[point_index][0],
-                            'y': sa_points[point_index][1]
+                            'x': x,
+                            'y': y
                         }
                     )