clean up

HumanSignal · Mar 28, 2023 · 9bd9d0f · 9bd9d0f
1 parent 1f9a30c
commit 9bd9d0f
Showing 1 changed file with 16 additions and 9 deletions.
diff --git a/label_studio_converter/converter.py b/label_studio_converter/converter.py
@@ -653,7 +653,12 @@ def add_image(images, width, height, image_id, image_path):
 
             for label in labels:
                 category_name = None
-                for key in ['rectanglelabels', 'polygonlabels', 'brushlabels', 'labels']:
+                for key in [
+                    'rectanglelabels',
+                    'polygonlabels',
+                    'brushlabels',
+                    'labels',
+                ]:
                     if key in label and len(label[key]) > 0:
                         category_name = label[key][0]
                         break
@@ -718,14 +723,16 @@ def add_image(images, width, height, image_id, image_path):
                     )
                 elif 'brushlabels' in label and brush.pycocotools_imported:
                     segmentation = brush.ls_rle_to_coco_rle(label["rle"], height, width)
-                    annotations.append({
-                        "image_id": image_id,
-                        "segmentation": segmentation,
-                        "area": brush.get_cocomask_area(segmentation),
-                        "bbox": brush.get_cocomask_bounding_box(segmentation),
-                        "iscrowd": 1,
-                        "category_id": category_id,
-                    })
+                    annotations.append(
+                        {
+                            "image_id": image_id,
+                            "segmentation": segmentation,
+                            "area": brush.get_cocomask_area(segmentation),
+                            "bbox": brush.get_cocomask_bounding_box(segmentation),
+                            "iscrowd": 1,
+                            "category_id": category_id,
+                        }
+                    )
                 else:
                     raise ValueError("Unknown label type")