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

fix: change None to list when HoughLinesP returns None

Browse Source
This commit is contained in:
iriley 2024-03-08 10:13:35 +01:00
parent ebdf3cefbf
commit ddd680bb4c
1 changed files with 10 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
src/cv_analysis/table_parsing.py
View File

@ -178,20 +178,17 @@ def detect_endpoints(
return (dist_a < 15) and (dist_b < 15) and overlap
lines = list(
map(
lambda x: tuple(x[0]),
cv2.HoughLinesP(
image, # Input edge image
1, # Distance resolution in pixels
np.pi / 180, # Angle resolution in radians
threshold=100, # Min number of votes for valid line
minLineLength=200, # Min allowed length of line
maxLineGap=10, # Max allowed gap between line for joining them
),
)
points = cv2.HoughLinesP(
image, # Input edge image
1, # Distance resolution in pixels
np.pi / 180, # Angle resolution in radians
threshold=100, # Min number of votes for valid line
minLineLength=200, # Min allowed length of line
maxLineGap=10, # Max allowed gap between line for joining them
)
points = points if points is not None else []
lines = list(map(lambda x: tuple(x[0]), points))
if not lines:
return lines