Skip to content

Commit 1db4655

Browse files
committed
fix: adjust merge row cell threhold
1 parent 2536129 commit 1db4655

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

tests/test_wired_table_rec.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ def test_squeeze_bug():
4141
ocr_result, _ = ocr_engine(img_path)
4242
table_str, *_ = table_recog(str(img_path), ocr_result)
4343
td_nums = get_td_nums(table_str)
44-
assert td_nums == 229
44+
assert td_nums == 228
4545

4646

4747
@pytest.mark.parametrize(

wired_table_rec/table_recover.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,7 @@ def get_benchmark_rows(
115115
leftmost_cell_idxs = [v[0] for v in rows.values()]
116116
benchmark_x = polygons[leftmost_cell_idxs][:, 0, 1]
117117

118-
theta = 10
118+
theta = 15
119119
# 遍历其他所有的框,按照y轴进行区间划分
120120
range_res = {}
121121
for cur_idx, cur_box in enumerate(polygons):

0 commit comments

Comments
 (0)