We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 13834be commit d8ea638Copy full SHA for d8ea638
1 file changed
mindee/parsing/standard/base.py
@@ -1,7 +1,6 @@
1
from typing import Any, List, Optional, Type
2
3
-from mindee.geometry.point import Point
4
-from mindee.geometry.polygon import Polygon
+from mindee.geometry.polygon import Polygon, polygon_from_prediction
5
from mindee.geometry.quadrilateral import Quadrilateral, get_bounding_box
6
from mindee.parsing.common.string_dict import StringDict
7
@@ -18,9 +17,7 @@ def _set_position(self, raw_prediction: StringDict):
18
17
self.bounding_box = None
19
self.polygon = Polygon()
20
try:
21
- self.polygon = Polygon(
22
- Point(point[0], point[1]) for point in raw_prediction["polygon"]
23
- )
+ self.polygon = polygon_from_prediction(raw_prediction["polygon"])
24
except (KeyError, TypeError):
25
pass
26
if self.polygon:
0 commit comments