mirrors
/
mmocr
mirror of https://github.com/open-mmlab/mmocr.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[Fix] Detect intersection before using shapley.intersection to eliminate spurious warnings (#1710) 

* [Fix] Detect intersection before using shapley.intersection to eliminate spurious warnings

* Update polygon_utils.py
pull/1712/head
Tong Gao 2023-02-08 10:36:11 +08:00 committed by GitHub
parent 2a2cab3c8c
commit d8e615921d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
mmocr/utils/polygon_utils.py
View File

@ -212,9 +212,8 @@ def poly_intersection(poly_a: Polygon,
float or tuple(float, Polygon): Returns the intersection area or
a tuple ``(area, Optional[poly_obj])``, where the `area` is the
intersection area between two polygons and `poly_obj` is The Polygon
object of the intersection area. Set as `None` if the input is invalid.
Set as `None` if the input is invalid. `poly_obj` will be returned
only if `return_poly` is `True`.
object of the intersection area, which will be `None` if the input is
invalid. `poly_obj` will be returned only if `return_poly` is `True`.
"""
assert isinstance(poly_a, Polygon)
assert isinstance(poly_b, Polygon)
@ -227,8 +226,12 @@ def poly_intersection(poly_a: Polygon,
poly_obj = None
area = invalid_ret
if poly_a.is_valid and poly_b.is_valid:
poly_obj = poly_a.intersection(poly_b)
area = poly_obj.area
if poly_a.intersects(poly_b):
poly_obj = poly_a.intersection(poly_b)
area = poly_obj.area
else:
poly_obj = Polygon()
area = 0.0
return (area, poly_obj) if return_poly else area