Fixed polygon detection

dhlab-epfl · Jan 6, 2021 · 799421d · 799421d
1 parent 29374bd
commit 799421d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dh_segment_torch/post_processing/geometries/polygon.py b/dh_segment_torch/post_processing/geometries/polygon.py
@@ -44,4 +44,4 @@ def apply(
         found_polygons = sorted(
             found_polygons, key=lambda poly: poly.area, reverse=True
         )
-        return found_polygons[: min(self.max_boxes, len(found_polygons))]
+        return found_polygons[: min(self.max_polygons, len(found_polygons))]