File tree Expand file tree Collapse file tree 1 file changed +2
-5
lines changed
Expand file tree Collapse file tree 1 file changed +2
-5
lines changed Original file line number Diff line number Diff line change 11from 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
54from mindee .geometry .quadrilateral import Quadrilateral , get_bounding_box
65from 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 :
You can’t perform that action at this time.
0 commit comments