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

Refactoring: Move

Browse Source
This commit is contained in:
Matthias Bisping 2023-01-09 17:01:36 +01:00
parent 619f67f1fd
commit c00081b2bc
2 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
cv_analysis/layout_parsing.py
View File

@ -3,10 +3,11 @@ from typing import Iterable
import cv2
import numpy as np
from funcy import compose, rcompose, first, lkeep
from funcy import compose, rcompose, lkeep
from cv_analysis.utils.merging import connect_related_rectangles
from cv_analysis.utils.common import find_contours
from cv_analysis.utils.conversion import box_to_rectangle
from cv_analysis.utils.merging import connect_related_rectangles
from cv_analysis.utils.postprocessing import remove_included, has_no_parent
from cv_analysis.utils.rectangle import Rectangle
@ -49,11 +50,6 @@ def __find_segments(image):
return rectangles
def find_contours(image):
contours, hierarchies = cv2.findContours(image, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
return contours, first(hierarchies)
def is_likely_segment(rect, min_area=100):
# FIXME: Parameterize via factory
return cv2.contourArea(rect, False) > min_area

7
cv_analysis/utils/common.py Normal file
View File

@ -0,0 +1,7 @@
import cv2
from funcy import first
def find_contours(image):
contours, hierarchies = cv2.findContours(image, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
return contours, first(hierarchies)