Marmelator/cv-analysis-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 37 Wiki Activity

Add typehints

Browse Source
This commit is contained in:
Matthias Bisping 2023-01-10 10:20:07 +01:00
parent 3cb857d830
commit 3134021596
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
cv_analysis/utils/merging.py
View File

@ -6,13 +6,13 @@ from cv_analysis.utils.rectangle import Rectangle
from cv_analysis.utils.spacial import is_related
def connect_related_rectangles(rectangles: List[Rectangle]):
def connect_related_rectangles(rectangles: List[Rectangle]) -> List[Rectangle]:
assert isinstance(rectangles, list)
no_new_merges = make_merger_sentinel()
return until(no_new_merges, merge_rectangles_once, rectangles)
def merge_rectangles_once(rectangles: List[Rectangle]):
def merge_rectangles_once(rectangles: List[Rectangle]) -> List[Rectangle]:
for alpha, beta in combinations(rectangles, 2):
if is_related(alpha, beta):
rectangles.remove(alpha)
@ -22,7 +22,7 @@ def merge_rectangles_once(rectangles: List[Rectangle]):
return rectangles
def bounding_rect(alpha: Rectangle, beta: Rectangle):
def bounding_rect(alpha: Rectangle, beta: Rectangle) -> Rectangle:
return Rectangle(
min(alpha.x1, beta.x1),
min(alpha.y1, beta.y1),