|
@@ -39,9 +39,10 @@ table_engine = PPStructure(layout=False,
|
|
|
table=True,
|
|
|
use_gpu=True,
|
|
|
show_log=True,
|
|
|
+ use_angle_cls=True,
|
|
|
# det_model_dir="models/det/det_table_v2",
|
|
|
# rec_model_dir="models/rec/rec_table_v1",
|
|
|
- table_model_dir="models/table/SLANet_905")
|
|
|
+ table_model_dir="models/table/SLANet_829")
|
|
|
|
|
|
|
|
|
cls_lock = threading.Lock()
|
|
@@ -142,6 +143,7 @@ def table_res(im, ROTATE=-1):
|
|
|
im = im.copy()
|
|
|
# 获取正向图片
|
|
|
img = get_zero_degree_image(im)
|
|
|
+ # cv2.imwrite('1.jpg', img)
|
|
|
try:
|
|
|
table_engine_lock.acquire()
|
|
|
res = table_engine(img)
|