chore: rm demo change

Author: Joker1212

demo_lineless.py

@@ -12,21 +12,21 @@
 
 output_dir = Path("outputs")
 output_dir.mkdir(parents=True, exist_ok=True)
-if __name__ == "__main__":
-    img_path = "tests/test_files/table.jpg"
-    table_rec = LinelessTableRecognition()
 
-    html, elasp, polygons, logic_points, ocr_res = table_rec(img_path)
-    print(f"cost: {elasp:.5f}")
+img_path = "tests/test_files/lineless_table_recognition.jpg"
+table_rec = LinelessTableRecognition()
 
-    complete_html = format_html(html)
+html, elasp, polygons, logic_points, ocr_res = table_rec(img_path)
+print(f"cost: {elasp:.5f}")
 
-    save_table_path = output_dir / "table.html"
-    with open(save_table_path, "w", encoding="utf-8") as file:
-        file.write(complete_html)
+complete_html = format_html(html)
 
-    plot_rec_box_with_logic_info(
-        img_path, f"{output_dir}/table_rec_box.jpg", logic_points, polygons
-    )
-    plot_rec_box(img_path, f"{output_dir}/ocr_box.jpg", ocr_res)
-    print(f"The results has been saved under {output_dir}")
+save_table_path = output_dir / "table.html"
+with open(save_table_path, "w", encoding="utf-8") as file:
+    file.write(complete_html)
+
+plot_rec_box_with_logic_info(
+    img_path, f"{output_dir}/table_rec_box.jpg", logic_points, polygons
+)
+plot_rec_box(img_path, f"{output_dir}/ocr_box.jpg", ocr_res)
+print(f"The results has been saved under {output_dir}")

demo_wired.py

@@ -15,21 +15,20 @@
 
 table_rec = WiredTableRecognition()
 
-img_path = "tests/test_files/wired/table_recognition.jpg"
-if __name__ == "__main__":
-    html, elasp, polygons, logic_points, ocr_res = table_rec(img_path)
+img_path = "tests/test_files/wired/table1.png"
+html, elasp, polygons, logic_points, ocr_res = table_rec(img_path)
 
-    print(f"cost: {elasp:.5f}")
+print(f"cost: {elasp:.5f}")
 
-    complete_html = format_html(html)
+complete_html = format_html(html)
 
-    save_table_path = output_dir / "table.html"
-    with open(save_table_path, "w", encoding="utf-8") as file:
-        file.write(complete_html)
+save_table_path = output_dir / "table.html"
+with open(save_table_path, "w", encoding="utf-8") as file:
+    file.write(complete_html)
 
-    plot_rec_box_with_logic_info(
-        img_path, f"{output_dir}/table_rec_box.jpg", logic_points, polygons
-    )
-    plot_rec_box(img_path, f"{output_dir}/ocr_box.jpg", ocr_res)
+plot_rec_box_with_logic_info(
+    img_path, f"{output_dir}/table_rec_box.jpg", logic_points, polygons
+)
+plot_rec_box(img_path, f"{output_dir}/ocr_box.jpg", ocr_res)
 
-    print(f"The results has been saved under {output_dir}")
+print(f"The results has been saved under {output_dir}")