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

[WIP] Refactoring meta-detection

Browse Source
This commit is contained in:
Matthias Bisping 2023-01-09 15:40:32 +01:00
parent 0f11441b20
commit acf46a7a48
2 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
cv_analysis/utils/connect_rects.py
View File

@ -3,16 +3,7 @@ from typing import List
from cv_analysis.utils import until, make_merger_sentinel from cv_analysis.utils import until, make_merger_sentinel
from cv_analysis.utils.rectangle import Rectangle from cv_analysis.utils.rectangle import Rectangle
from cv_analysis.utils.spacial import overlap, adjacent from cv_analysis.utils.spacial import is_related
def close(alpha: Rectangle, beta: Rectangle, max_gap=14):
# FIXME: Parameterize via factory
return adjacent(alpha, beta, tolerance=max_gap, strict=True)
def is_related(alpha: Rectangle, beta: Rectangle):
return close(alpha, beta) or overlap(alpha, beta)
def connect_related_rectangles(rectangles: List[Rectangle]): def connect_related_rectangles(rectangles: List[Rectangle]):

9
cv_analysis/utils/spacial.py
View File

@ -273,5 +273,14 @@ def intersection_along_axis(alpha, beta, axis):
return intersection return intersection
def is_related(alpha: Rectangle, beta: Rectangle):
return close(alpha, beta) or overlap(alpha, beta)
def close(alpha: Rectangle, beta: Rectangle, max_gap=14):
# FIXME: Parameterize via factory
return adjacent(alpha, beta, tolerance=max_gap, strict=True)
def overlap(alpha: Rectangle, beta: Rectangle): def overlap(alpha: Rectangle, beta: Rectangle):
return intersection(alpha, beta) > 0 return intersection(alpha, beta) > 0