Skip to content

Commit d8ea638

Browse files
committed
♻️ better polygon init in standard field
1 parent 13834be commit d8ea638

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

mindee/parsing/standard/base.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
from typing import Any, List, Optional, Type
22

3-
from mindee.geometry.point import Point
4-
from mindee.geometry.polygon import Polygon
3+
from mindee.geometry.polygon import Polygon, polygon_from_prediction
54
from mindee.geometry.quadrilateral import Quadrilateral, get_bounding_box
65
from mindee.parsing.common.string_dict import StringDict
76

@@ -18,9 +17,7 @@ def _set_position(self, raw_prediction: StringDict):
1817
self.bounding_box = None
1918
self.polygon = Polygon()
2019
try:
21-
self.polygon = Polygon(
22-
Point(point[0], point[1]) for point in raw_prediction["polygon"]
23-
)
20+
self.polygon = polygon_from_prediction(raw_prediction["polygon"])
2421
except (KeyError, TypeError):
2522
pass
2623
if self.polygon:

0 commit comments

Comments
 (0)