Filter out very small regions in segmenter

mittagessen · Mar 6, 2024 · 4248ca0 · 4248ca0
1 parent feeecc9
commit 4248ca0
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/kraken/lib/segmentation.py b/kraken/lib/segmentation.py
@@ -352,10 +352,13 @@ def vectorize_regions(im: np.ndarray, threshold: float = 0.5):
         [[x0, y0, ... xn, yn], [xm, ym, ..., xk, yk], ... ]
         A list of lists containing the region polygons.
     """
+    print(f'shape: {im.shape} {im.max()}')
     bin = im > threshold
     labelled = label(bin)
     boundaries = []
     for x in regionprops(labelled):
+        if x.area < 32:
+            continue
         boundary = boundary_tracing(x)
         if len(boundary) > 2:
             boundaries.append(geom.Polygon(boundary))