浏览代码

Merge pull request #2914 from myhloli/dev

dev
Xiaomeng Zhao 4 月之前
父节点
当前提交
53f9b9c6d5
共有 1 个文件被更改,包括 11 次插入8 次删除
  1. 11 8
      mineru/model/table/rapid_table.py

+ 11 - 8
mineru/model/table/rapid_table.py

@@ -76,11 +76,14 @@ class RapidTableModel(object):
 
 
         if ocr_result:
-            table_results = self.table_model(np.asarray(image), ocr_result)
-            html_code = table_results.pred_html
-            table_cell_bboxes = table_results.cell_bboxes
-            logic_points = table_results.logic_points
-            elapse = table_results.elapse
-            return html_code, table_cell_bboxes, logic_points, elapse
-        else:
-            return None, None, None, None
+            try:
+                table_results = self.table_model(np.asarray(image), ocr_result)
+                html_code = table_results.pred_html
+                table_cell_bboxes = table_results.cell_bboxes
+                logic_points = table_results.logic_points
+                elapse = table_results.elapse
+                return html_code, table_cell_bboxes, logic_points, elapse
+            except Exception as e:
+                logger.exception(e)
+
+        return None, None, None, None