修Bug：引擎原始输出项的"text"为空时，导致文本分析的越界错误 (#625)

hiroi-sora · Aug 16, 2024 · cd7974f · cd7974f
1 parent f601c3c
commit cd7974f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/UmiOCR-data/py_src/ocr/tbpu/parser_tools/line_preprocessing.py b/UmiOCR-data/py_src/ocr/tbpu/parser_tools/line_preprocessing.py
@@ -81,8 +81,9 @@ def _getBboxes(textBlocks, rotation_rad):
     return bboxes
 
 
-# 预处理 textBlocks ，将包围盒 ["box"] 转为标准化 bbox
+# 预处理 textBlocks ，将包围盒 ["box"] 转为标准化 bbox ，同时去除 ["text"] 不完整的项
 def linePreprocessing(textBlocks):
+    textBlocks = [i for i in textBlocks if i.get("text", False)]
     # 判断角度
     rotation_rad = _estimateRotation(textBlocks)
     # 获取标准化bbox