From 2e2571cdc2e4768ee01248492096b059ab25da5d Mon Sep 17 00:00:00 2001 From: Joker1212 <519548295@qq.com> Date: Thu, 12 Dec 2024 15:06:41 +0800 Subject: [PATCH] fix: add default value for enhance_box_line --- README.md | 2 ++ wired_table_rec/table_line_rec_plus.py | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 51ede8e..7dfa466 100644 --- a/README.md +++ b/README.md @@ -69,6 +69,8 @@ ### 使用建议 wired_table_rec_v2(有线表格精度最高): 通用场景有线表格(论文,杂志,期刊, 收据,单据,账单) +wired_table_rec_v2 对1500px内大小的图片效果最好,所以分辨率超过2000px建议等比缩放一下 + paddlex-SLANet-plus(综合精度最高): 文档场景表格(论文,杂志,期刊中的表格) ### 安装 diff --git a/wired_table_rec/table_line_rec_plus.py b/wired_table_rec/table_line_rec_plus.py index def5021..38d282c 100644 --- a/wired_table_rec/table_line_rec_plus.py +++ b/wired_table_rec/table_line_rec_plus.py @@ -81,7 +81,7 @@ def postprocess(self, img, pred, **kwargs): h_lines_threshold = kwargs.get("h_lines_threshold", 100) if kwargs else 100 v_lines_threshold = kwargs.get("v_lines_threshold", 15) if kwargs else 15 angle = kwargs.get("angle", 50) if kwargs else 50 - enhance_box_line = kwargs.get("enhance_box_line") if kwargs else True + enhance_box_line = kwargs.get("enhance_box_line", True) if kwargs else True morph_close = ( kwargs.get("morph_close", enhance_box_line) if kwargs else enhance_box_line ) # 是否进行闭合运算以找到更多小的框