瀏覽代碼

feat(wired_table): 添加多源单元格融合模块到有线表格识别子模块

zhch158_admin 2 周之前
父節點
當前提交
75b1d72e9c
共有 1 個文件被更改,包括 3 次插入0 次删除
  1. 3 0
      ocr_tools/universal_doc_parser/models/adapters/wired_table/__init__.py

+ 3 - 0
ocr_tools/universal_doc_parser/models/adapters/wired_table/__init__.py

@@ -9,6 +9,7 @@
 - 可视化
 - OCR格式转换
 - 调试工具
+- 多源单元格融合(UNet + RT-DETR + OCR)
 """
 
 from .debug_utils import WiredTableDebugOptions, WiredTableDebugUtils
@@ -18,6 +19,7 @@ from .grid_recovery import GridRecovery
 from .text_filling import TextFiller
 from .html_generator import WiredTableHTMLGenerator
 from .visualization import WiredTableVisualizer
+from .cell_fusion import CellFusionEngine
 
 __all__ = [
     'WiredTableDebugOptions',
@@ -28,5 +30,6 @@ __all__ = [
     'TextFiller',
     'WiredTableHTMLGenerator',
     'WiredTableVisualizer',
+    'CellFusionEngine',
 ]