Skip to content

GitLab

Unverified Commit 53f9b9c6 authored by Xiaomeng Zhao's avatar Xiaomeng Zhao Committed by GitHub
Browse files

Merge pull request #2914 from myhloli/dev 

dev
parents b2d11663 3d3bdf0b
Show whitespace changes
Inline Side-by-side
Showing with 11 additions and 8 deletions
mineru/model/table/rapid_table.py
View file @ 53f9b9c6
...@@ -76,11 +76,14 @@ class RapidTableModel(object): ...@@ -76,11 +76,14 @@ class RapidTableModel(object):
if ocr_result: if ocr_result:
try:
table_results = self.table_model(np.asarray(image), ocr_result) table_results = self.table_model(np.asarray(image), ocr_result)
html_code = table_results.pred_html html_code = table_results.pred_html
table_cell_bboxes = table_results.cell_bboxes table_cell_bboxes = table_results.cell_bboxes
logic_points = table_results.logic_points logic_points = table_results.logic_points
elapse = table_results.elapse elapse = table_results.elapse
return html_code, table_cell_bboxes, logic_points, elapse return html_code, table_cell_bboxes, logic_points, elapse
else: except Exception as e:
logger.exception(e)
return None, None, None, None return None, None, None, None
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment