From 799421d7bf8bc9e5bf592036696f057e8073a87d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Rapha=C3=ABl=20Barman?= Date: Wed, 6 Jan 2021 14:23:24 +0100 Subject: [PATCH] Fixed polygon detection --- dh_segment_torch/post_processing/geometries/polygon.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dh_segment_torch/post_processing/geometries/polygon.py b/dh_segment_torch/post_processing/geometries/polygon.py index 07c3aec..db94713 100644 --- 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))]