Linter

Author: feynmanliang

labelbox/exporters/coco_exporter.py

@@ -5,13 +5,13 @@
 import datetime as dt
 import json
 import logging
-from PIL import Image
 import requests
 from shapely import wkt
 from shapely.geometry import Polygon
 from typing import Any, Dict, Sequence
 
 from labelbox.exceptions import UnknownFormatError
+from PIL import Image
 
 
 def from_json(labeled_data, coco_output, label_format='WKT'):
@@ -78,7 +78,8 @@ def add_label(
         label_id: ID for the instance to write
         image_url: URL to download image file from
         labels: Labelbox formatted labels to use for generating annotation
-        label_format: Format of the labeled data. Valid options are: "WKT" and "XY", default is "WKT".
+        label_format: Format of the labeled data. Valid options are: "WKT" and
+                      "XY", default is "WKT".
 
     Returns:
         The updated COCO export represented as a dictionary.
@@ -118,25 +119,29 @@ def add_label(
             coco['categories'].append(category)
 
         polygons = _get_polygons(label_format, label_data)
-
-        for polygon in polygons:
-            segmentation = []
-            for x_val, y_val in polygon.exterior.coords:
-                segmentation.extend([x_val, image['height'] - y_val])
-
-            annotation = {
-                "id": len(coco['annotations']) + 1,
-                "image_id": image['id'],
-                "category_id": category_id,
-                "segmentation": [segmentation],
-                "area": polygon.area,  # float
-                "bbox": [polygon.bounds[0], polygon.bounds[1],
-                         polygon.bounds[2] - polygon.bounds[0],
-                         polygon.bounds[3] - polygon.bounds[1]],
-                "iscrowd": 0
-            }
-
-            coco['annotations'].append(annotation)
+        _append_polygons_as_annotations(coco, image, category_id, polygons)
+
+
+def _append_polygons_as_annotations(coco, image, category_id, polygons):
+    "Adds `polygons` as annotations in the `coco` export"
+    for polygon in polygons:
+        segmentation = []
+        for x_val, y_val in polygon.exterior.coords:
+            segmentation.extend([x_val, image['height'] - y_val])
+
+        annotation = {
+            "id": len(coco['annotations']) + 1,
+            "image_id": image['id'],
+            "category_id": category_id,
+            "segmentation": [segmentation],
+            "area": polygon.area,  # float
+            "bbox": [polygon.bounds[0], polygon.bounds[1],
+                     polygon.bounds[2] - polygon.bounds[0],
+                     polygon.bounds[3] - polygon.bounds[1]],
+            "iscrowd": 0
+        }
+
+        coco['annotations'].append(annotation)
 
 
 def _get_polygons(label_format, label_data):

labelbox/exporters/voc_exporter.py

@@ -5,13 +5,13 @@
 import json
 import logging
 import os
-from PIL import Image
 import requests
 from shapely import wkt
 from typing import Any, Sequence
 
 from labelbox.exceptions import UnknownFormatError
 from labelbox.exporters.pascal_voc_writer import Writer as PascalWriter
+from PIL import Image
 
 
 def from_json(labeled_data, annotations_output_dir, images_output_dir,
@@ -70,9 +70,10 @@ def write_label(
         label_id: ID for the instance to write
         image_url: URL to download image file from
         labels: Labelbox formatted labels to use for generating annotation
-        label_format: Format of the labeled data. Valid options are: "WKT" and "XY", default is "WKT".
+        label_format: Format of the labeled data. Valid options are: "WKT" and
+                      "XY", default is "WKT".
         annotations_output_dir: File path of directory to write Pascal VOC
-            annotation files.
+                                annotation files.
         images_output_dir: File path of directory to write images.
     """
     # Download image and save it