removed annotations of zero area

waspinator · Apr 10, 2018 · 38ee683 · 38ee683
1 parent 4511a15
commit 38ee683
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pycococreatortools/pycococreatortools.py b/pycococreatortools/pycococreatortools.py
@@ -78,6 +78,9 @@ def create_annotation_info(annotation_id, image_id, category_info, binary_mask,
     bounding_box = mask.toBbox(binary_mask_encoded)
     area = mask.area(binary_mask_encoded)
 
+    if area < 1:
+        return None
+
     if category_info["is_crowd"]:
         annotation_info = {
             "id": annotation_id,