Fixed polygon detection

Raphaël Barman · Raphaël Barman · commit 799421d7bf8b · 2021-01-06T14:23:24.000+01:00
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))]

Original file line number	Diff line number	Diff line change
`@@ -44,4 +44,4 @@ def apply(`
`44`	`44`	`found_polygons = sorted(`
`45`	`45`	`found_polygons, key=lambda poly: poly.area, reverse=True`
`46`	`46`	`)`
`47`		`- return found_polygons[: min(self.max_boxes, len(found_polygons))]`
	`47`	`+ return found_polygons[: min(self.max_polygons, len(found_polygons))]`