Merge pull request #72 from albu/fix_bbox_example

fix bbox example and bug with filtering
albumentations-team · Sep 20, 2018 · 9717919 · 9717919
2 parents 9e027f5 + 6635fda
commit 9717919
Show file tree

Hide file tree

Showing 3 changed files with 101 additions and 131 deletions.
diff --git a/albumentations/__init__.py b/albumentations/__init__.py
@@ -1,6 +1,6 @@
 from __future__ import absolute_import
 
-__version__ = '0.0.15'
+__version__ = '0.0.16'
 
 from .core.composition import *
 from .core.transforms_interface import *

diff --git a/albumentations/augmentations/bbox_utils.py b/albumentations/augmentations/bbox_utils.py
@@ -204,15 +204,14 @@ def filter_bboxes(bboxes, rows, cols, min_area=0., min_visibility=0.):
     """
     resulting_boxes = []
     for bbox in bboxes:
-        if min_visibility:
-            transformed_box_area = calculate_bbox_area(bbox, rows, cols)
-            bbox[:4] = np.clip(bbox[:4], 0, 1.)
-            clipped_box_area = calculate_bbox_area(bbox, rows, cols)
-            if not transformed_box_area or clipped_box_area / transformed_box_area < min_visibility:
-                continue
+        transformed_box_area = calculate_bbox_area(bbox, rows, cols)
+        bbox[:4] = np.clip(bbox[:4], 0, 1.)
+        clipped_box_area = calculate_bbox_area(bbox, rows, cols)
+        if not transformed_box_area or clipped_box_area / transformed_box_area <= min_visibility:
+            continue
         else:
             bbox[:4] = np.clip(bbox[:4], 0, 1.)
-        if min_area and calculate_bbox_area(bbox, rows, cols) < min_area:
+        if calculate_bbox_area(bbox, rows, cols) <= min_area:
             continue
         resulting_boxes.append(bbox)
     return resulting_boxes
diff --git a/notebooks/example_bboxes.ipynb b/notebooks/example_bboxes.ipynb