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

Merge branch 'table_lines' into 'master'

Browse Source 
fix: check nonzero list length in filter_fp_col_lines

See merge request redactmanager/cv-analysis-service!21
This commit is contained in:
Isaac Riley 2024-05-13 09:39:20 +02:00
commit 3c9ddfcf0f
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pyproject.toml
View File

@ -1,6 +1,6 @@
[tool.poetry] [tool.poetry]
name = "cv-analysis-service" name = "cv-analysis-service"
version = "2.15.1" version = "2.16.1"
description = "" description = ""
authors = ["Isaac Riley <isaac.riley@knecon.com>"] authors = ["Isaac Riley <isaac.riley@knecon.com>"]
readme = "README.md" readme = "README.md"

4
src/cv_analysis/table_inference.py
View File

@ -135,9 +135,11 @@ FILTERS: dict[str, dict[int, np.ndarray | None]] = {
def filter_fp_col_lines(line_list: list[int], filt_sums: Array) -> list[int]: def filter_fp_col_lines(line_list: list[int], filt_sums: Array) -> list[int]:
if not line_list: if not list(line_list):
return [] return []
centers = list(np.where((filt_sums[1:-1] < filt_sums[:-2]) * (filt_sums[1:-1] < filt_sums[2:]))[0] + 1) centers = list(np.where((filt_sums[1:-1] < filt_sums[:-2]) * (filt_sums[1:-1] < filt_sums[2:]))[0] + 1)
if not centers:
return []
if line_list[0] > centers[0]: if line_list[0] > centers[0]:
centers = centers[1:] + [len(filt_sums) - 1] centers = centers[1:] + [len(filt_sums) - 1]