update app

app.py

@@ -1,3 +1,4 @@
+import threading
 import time
 
 import cv2
@@ -6,10 +7,11 @@ from lineless_table_rec import LinelessTableRecognition
 from paddleocr import PPStructure
 from rapid_table import RapidTable
 from rapidocr_onnxruntime import RapidOCR
+from slanet_plus_table import SLANetPlus
 from table_cls import TableCls
 from wired_table_rec import WiredTableRecognition
-from utils import plot_rec_box, LoadImage, format_html, box_4_2_poly_to_box_4_1
 
+from utils import plot_rec_box, LoadImage, format_html, box_4_2_poly_to_box_4_1
 img_loader = LoadImage()
 table_rec_path = "models/table_rec/ch_ppstructure_mobile_v2_SLANet.onnx"
 det_model_dir = {
@@ -21,7 +23,8 @@ rec_model_dir = {
 }
 table_engine_list = [
     "auto",
-    "rapid_table",
+    "RapidTable(SLANet)",
+    "RapidTable(SLANet-plus)",
     "wired_table_v2",
     "pp_table",
     "wired_table_v1",
@@ -41,6 +44,7 @@ example_images = [
     "images/wired6.jpg",
 ]
 rapid_table_engine = RapidTable(model_path=table_rec_path)
+SLANet_plus_table_Engine = RapidTable()
 wired_table_engine_v1 = WiredTableRecognition(version="v1")
 wired_table_engine_v2 = WiredTableRecognition(version="v2")
 lineless_table_engine = LinelessTableRecognition()
@@ -69,8 +73,10 @@ def select_ocr_model(det_model, rec_model):
 
 
 def select_table_model(img, table_engine_type, det_model, rec_model):
-    if table_engine_type == "rapid_table":
+    if table_engine_type == "RapidTable(SLANet)":
         return rapid_table_engine, table_engine_type
+    elif table_engine_type == "RapidTable(SLANet-plus)":
+        return SLANet_plus_table_Engine, table_engine_type
     elif table_engine_type == "wired_table_v1":
         return wired_table_engine_v1, table_engine_type
     elif table_engine_type == "wired_table_v2":
@@ -105,7 +111,6 @@ def process_image(img, table_engine_type, det_model, rec_model):
         ocr_res, ocr_infer_elapse = ocr_engine(img)
         det_cost, cls_cost, rec_cost = ocr_infer_elapse
         ocr_boxes = [box_4_2_poly_to_box_4_1(ori_ocr[0]) for ori_ocr in ocr_res]
-
         if isinstance(table_engine, RapidTable):
             html, polygons, table_rec_elapse = table_engine(img, ocr_result=ocr_res)
             polygons = [[polygon[0], polygon[1], polygon[4], polygon[5]] for polygon in polygons]
@@ -176,4 +181,4 @@ def main():
 
 
 if __name__ == '__main__':
-    main()
+    main()