test: add test for new param

Author: Joker1212

demo_wired.py

@@ -15,17 +15,8 @@
 
 table_rec = WiredTableRecognition()
 
-img_path = "tests/test_files/wired/wired_big_box.png"
-html, elasp, polygons, logic_points, ocr_res = table_rec(
-    img_path,
-    version="v2",  # 默认使用v2线框模型，切换阿里读光模型可改为v1
-    morph_close=True,  # 是否进行形态学操作,辅助找到更多线框,默认为True
-    more_h_lines=True,  # 是否基于线框检测结果进行更多水平线检查，辅助找到更小线框, 默认为True
-    more_v_lines=True,  # 是否基于线框检测结果进行更多垂直线检查，辅助找到更小线框, 默认为True
-    extend_line=True,  # 是否基于线框检测结果进行线段延长，辅助找到更多线框, 默认为True
-    need_ocr=True,  # 是否进行OCR识别, 默认为True
-    rec_again=True,  # 是否针对未识别到文字的表格框,进行单独截取再识别,默认为True
-)
+img_path = "tests/test_files/wired/table1.png"
+html, elasp, polygons, logic_points, ocr_res = table_rec(img_path)
 
 print(f"cost: {elasp:.5f}")
 
@@ -38,6 +29,6 @@
 plot_rec_box_with_logic_info(
     img_path, f"{output_dir}/table_rec_box.jpg", logic_points, polygons
 )
-plot_rec_box(f"{output_dir}/table_rec_box.jpg", f"{output_dir}/ocr_box.jpg", ocr_res)
+plot_rec_box(img_path, f"{output_dir}/ocr_box.jpg", ocr_res)
 
 print(f"The results has been saved under {output_dir}")

wired_table_rec/utils_table_recover.py

@@ -267,17 +267,8 @@ def plot_rec_box_with_logic_info(img_path, output_path, logic_points, sorted_pol
 
         cv2.putText(
             img,
-            f"row:{logic_points[idx][0]}-{logic_points[idx][1]}",
-            (x0 + 1, y0 + 15),
-            cv2.FONT_HERSHEY_PLAIN,
-            font_scale,
-            (0, 0, 255),
-            thickness,
-        )
-        cv2.putText(
-            img,
-            f"col:{logic_points[idx][2]}-{logic_points[idx][3]}",
-            (x0 + 1, y0 + 40),
+            f"{idx}-{logic_points[idx]}",
+            (x1, y1),
             cv2.FONT_HERSHEY_PLAIN,
             font_scale,
             (0, 0, 255),
@@ -312,15 +303,15 @@ def plot_rec_box(img_path, output_path, sorted_polygons):
         font_scale = 1.0  # 原先是0.5
         thickness = 2  # 原先是1
 
-        # cv2.putText(
-        #     img,
-        #     str(idx),
-        #     (x1, y1),
-        #     cv2.FONT_HERSHEY_PLAIN,
-        #     font_scale,
-        #     (0, 0, 255),
-        #     thickness,
-        # )
+        cv2.putText(
+            img,
+            str(idx),
+            (x1, y1),
+            cv2.FONT_HERSHEY_PLAIN,
+            font_scale,
+            (0, 0, 255),
+            thickness,
+        )
     os.makedirs(os.path.dirname(output_path), exist_ok=True)
     # 保存绘制后的图像
     cv2.imwrite(output_path, img)