Przeglądaj źródła

refactor: improve clarity of instructions in README_zh-CN

myhloli 4 miesięcy temu
rodzic
commit
3d3bdf0bea
1 zmienionych plików z 11 dodań i 8 usunięć
  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