Fix coco-convert-empty-array

Author: dshabin

superannotate/input_converters/converters/coco_converters/coco_api.py

@@ -103,6 +103,8 @@ def _area(bitmask):
 
 def _toBbox(bitmask):
     y, x = np.where(bitmask)
+    if x.tolist() == [] and y.tolist() == []:
+        return [0, 0, 0, 0]
     xmin = int(min(x))
     xmax = int(max(x))
     ymin = int(min(y))

tests/converter_test/COCO/input/fromSuperAnnotate/cats_dogs_vector_instance_segm_empty_array/classes/classes.json

@@ -0,0 +1,12 @@
+[
+    {
+        "id": 1,
+        "project_id": 1,
+        "name": "class1",
+        "color": "#626bef",
+        "count": 0,
+        "createdAt": "2021-03-31T21:13:10.000Z",
+        "updatedAt": "2021-03-31T21:13:10.000Z",
+        "attribute_groups": []
+    }
+]

tests/converter_test/COCO/input/fromSuperAnnotate/cats_dogs_vector_instance_segm_empty_array/some_image.JPG___objects.json

@@ -0,0 +1,53 @@
+{
+    "metadata": {
+        "lastAction": {
+            "email": "",
+            "timestamp": 1619092786567
+        },
+        "width": 2448,
+        "height": 3264,
+        "name": "q.JPG",
+        "projectId": 1,
+        "isPredicted": false,
+        "status": "Completed",
+        "pinned": false,
+        "annotatorEmail": "",
+        "qaEmail": ""
+    },
+    "comments": [],
+    "tags": [],
+    "instances": [
+        {
+            "type": "polygon",
+            "classId": 1,
+            "probability": 100,
+            "points": [
+                2448,
+                2612.07,
+                2448,
+                2651.45,
+                2448,
+                988.83
+            ],
+            "groupId": 0,
+            "pointLabels": {},
+            "locked": false,
+            "visible": true,
+            "attributes": [],
+            "trackingId": null,
+            "error": null,
+            "createdAt": "2021-04-21T06:56:10.242Z",
+            "createdBy": {
+                "email": "",
+                "role": ""
+            },
+            "creationType": "Manual",
+            "updatedAt": "2021-04-21T06:56:10.242Z",
+            "updatedBy": {
+                "email": "",
+                "role": ""
+            },
+            "className": "class1"
+        }
+    ]
+}

tests/converter_test/COCO/input/fromSuperAnnotate/cats_dogs_vector_instance_segm_empty_array/truth.json

@@ -0,0 +1 @@
+{"info": {"description": "This is instance_test_vector dataset.", "url": "https://superannotate.ai", "version": "1.0", "year": 2021, "contributor": "Superannotate AI", "date_created": "17/05/2021"}, "licenses": [{"url": "https://superannotate.ai", "id": 1, "name": "Superannotate AI"}], "images": [{"id": 1, "file_name": "q.JPG", "height": 3264, "width": 2448, "license": 1}], "annotations": [{"id": 1, "image_id": 1, "segmentation": [[2448, 2612.07, 2448, 2651.45, 2448, 988.83]], "iscrowd": 0, "bbox": [0, 0, 0, 0], "area": 0, "category_id": 1}], "categories": [{"id": 1, "name": "class1", "supercategory": "class1", "isthing": 1, "color": [1, 0, 0]}]}

tests/converter_test/test_conversion.py

@@ -94,3 +94,22 @@ def test_instance_segmentation_sa2coco_vector(tmpdir):
         input_dir, out_path, "COCO", "instance_test_vector", "Vector",
         "instance_segmentation"
     )
+
+
+def test_instance_segmentation_sa2coco_vector_empty_array(tmpdir):
+    input_dir = Path(
+        "tests"
+    ) / "converter_test" / "COCO" / "input" / "fromSuperAnnotate" / "cats_dogs_vector_instance_segm_empty_array"
+    out_path = Path(tmpdir) / "fromSuperAnnotate" / "instance_test_vector"
+    sa.export_annotation(
+        input_dir, out_path, "COCO", "instance_test_vector", "Vector",
+        "instance_segmentation"
+    )
+    import json
+    json_path = out_path / "instance_test_vector.json"
+    with open(json_path, 'r') as f:
+        data = json.loads(f.read())
+    truth_path = input_dir / "truth.json"
+    with open(truth_path, 'r') as f:
+        truth = json.loads(f.read())
+    assert truth == data