From 38ee683ac83f3fcec05b1225396af3b7b9606acd Mon Sep 17 00:00:00 2001 From: waspinator Date: Tue, 10 Apr 2018 14:05:07 -0400 Subject: [PATCH] removed annotations of zero area --- pycococreatortools/pycococreatortools.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/pycococreatortools/pycococreatortools.py b/pycococreatortools/pycococreatortools.py index 3a20e9c..8fd9c25 100644 --- 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,