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

Add fixmes and format docstring

Browse Source
This commit is contained in:
Matthias Bisping 2023-01-10 09:39:10 +01:00
parent 77f85e9de1
commit 5d1d9516b5
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
cv_analysis/figure_detection/figures.py
View File

@ -5,22 +5,28 @@ from cv_analysis.utils.common import find_contours
def detect_large_coherent_structures(image: np.array):
"""Detects large coherent structures on an image.
"""Detects large coherent structures in an image.
Expects an image with binary color space (e.g. threshold applied).
Args:
image (np.array): Image to look for large coherent structures in.
Returns:
contours
list: List of contours.
References:
https://stackoverflow.com/questions/60259169/how-to-group-nearby-contours-in-opencv-python-zebra-crossing-detection
"""
assert len(image.shape) == 2
# FIXME: Parameterize via factory
dilate_kernel = cv2.getStructuringElement(cv2.MORPH_OPEN, (5, 5))
# FIXME: Parameterize via factory
dilate = cv2.dilate(image, dilate_kernel, iterations=4)
# FIXME: Parameterize via factory
close_kernel = cv2.getStructuringElement(cv2.MORPH_RECT, (20, 20))
close = cv2.morphologyEx(dilate, cv2.MORPH_CLOSE, close_kernel, iterations=1)
# FIXME: Parameterize via factory
close = cv2.morphologyEx(dilate, cv2.MORPH_CLOSE, close_kernel, iterations=1) # TODO: Tweak iterations
contours, _ = find_contours(close)