|
|
@@ -27,7 +27,7 @@ app.add_middleware(
|
|
|
use_gpu = os.getenv('USE_CUDA') == 'gpu'
|
|
|
print(f'use gpu: {use_gpu}')
|
|
|
|
|
|
-# 初始化模型
|
|
|
+# 普通表格
|
|
|
table_engine = PPStructure(layout=False,
|
|
|
table=True,
|
|
|
use_gpu=use_gpu,
|
|
|
@@ -36,6 +36,7 @@ table_engine = PPStructure(layout=False,
|
|
|
rec_model_dir="./models/rec/rec_table_v1",
|
|
|
table_model_dir="models/table/SLAnet_v1")
|
|
|
|
|
|
+# 长度较长表格
|
|
|
table_engine1 = PPStructure(layout=False,
|
|
|
table=True,
|
|
|
use_gpu=use_gpu,
|
|
|
@@ -44,6 +45,7 @@ table_engine1 = PPStructure(layout=False,
|
|
|
rec_model_dir="./models/rec/rec_table_v1",
|
|
|
table_model_dir="./models/table/SLAnet_v1")
|
|
|
|
|
|
+# 针对某些特殊情况的补充模型
|
|
|
table_engine2 = PPStructure(layout=False,
|
|
|
table=True,
|
|
|
use_gpu=use_gpu,
|
