Parcourir la source

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

zhch158_admin il y a 2 semaines
Parent
commit
75b1d72e9c

+ 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',
 ]