Skip to content

Commit f492e8f

Browse files
authored
Merge pull request #131 from xinzheng99/main
fix: keyword errors & missing import
2 parents 1861826 + eb0209a commit f492e8f

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

README.md

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -89,6 +89,7 @@ from lineless_table_rec import LinelessTableRecognition
8989
from lineless_table_rec.utils_table_recover import format_html, plot_rec_box_with_logic_info, plot_rec_box
9090
from table_cls import TableCls
9191
from wired_table_rec import WiredTableRecognition
92+
from rapidocr_onnxruntime import RapidOCR
9293

9394
lineless_engine = LinelessTableRecognition()
9495
wired_engine = WiredTableRecognition()
@@ -106,7 +107,7 @@ html, elasp, polygons, logic_points, ocr_res = table_engine(img_path)
106107
print(f"elasp: {elasp}")
107108

108109
# 使用其他ocr模型
109-
#ocr_engine =RapidOCR(det_model_dir="xxx/det_server_infer.onnx",rec_model_dir="xxx/rec_server_infer.onnx")
110+
#ocr_engine =RapidOCR(det_model_path="xxx/det_server_infer.onnx",rec_model_path="xxx/rec_server_infer.onnx")
110111
#ocr_res, _ = ocr_engine(img_path)
111112
#html, elasp, polygons, logic_points, ocr_res = table_engine(img_path, ocr_result=ocr_res)
112113
# output_dir = f'outputs'

README_en.md

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -87,6 +87,7 @@ from lineless_table_rec import LinelessTableRecognition
8787
from lineless_table_rec.utils_table_recover import format_html, plot_rec_box_with_logic_info, plot_rec_box
8888
from table_cls import TableCls
8989
from wired_table_rec import WiredTableRecognition
90+
from rapidocr_onnxruntime import RapidOCR
9091

9192
lineless_engine = LinelessTableRecognition()
9293
wired_engine = WiredTableRecognition()
@@ -104,7 +105,7 @@ html, elasp, polygons, logic_points, ocr_res = table_engine(img_path)
104105
print(f"elasp: {elasp}")
105106

106107
# Use other OCR models
107-
#ocr_engine =RapidOCR(det_model_dir="xxx/det_server_infer.onnx",rec_model_dir="xxx/rec_server_infer.onnx")
108+
#ocr_engine =RapidOCR(det_model_path="xxx/det_server_infer.onnx",rec_model_path="xxx/rec_server_infer.onnx")
108109
#ocr_res, _ = ocr_engine(img_path)
109110
#html, elasp, polygons, logic_points, ocr_res = table_engine(img_path, ocr_result=ocr_res)
110111

0 commit comments

Comments
 (0)