|
@@ -31,11 +31,15 @@ if os.getenv('USE_CUDA') == 'gpu':
|
|
|
print(f'use gpu: {use_gpu}')
|
|
print(f'use gpu: {use_gpu}')
|
|
|
|
|
|
|
|
# 初始化模型
|
|
# 初始化模型
|
|
|
|
|
+<<<<<<< HEAD
|
|
|
|
|
+table_engine = PPStructure(layout=False, table=True, show_log=True, table_model_dir="/Users/sxkj/opt/python-workspace/yili-ocr/ocr-table/SLANet/")
|
|
|
|
|
+=======
|
|
|
table_engine = PPStructure(layout=False,
|
|
table_engine = PPStructure(layout=False,
|
|
|
table=True,
|
|
table=True,
|
|
|
use_gpu=use_gpu,
|
|
use_gpu=use_gpu,
|
|
|
show_log=True,
|
|
show_log=True,
|
|
|
table_model_dir="./tabel_ocr_infer")
|
|
table_model_dir="./tabel_ocr_infer")
|
|
|
|
|
+>>>>>>> d046d2d76465ba7ba4d8c3e2cdad1ec5e7ac3b69
|
|
|
|
|
|
|
|
|
|
|
|
|
class TableInfo(BaseModel):
|
|
class TableInfo(BaseModel):
|
liutao