Просмотр исходного кода

feat: 添加创建表格分类器的方法以区分有线/无线表格

zhch158_admin 2 дней назад
Родитель
Сommit
57bcb4628d
1 измененных файлов с 13 добавлено и 0 удалено
  1. 13 0
      ocr_tools/universal_doc_parser/core/model_factory.py

+ 13 - 0
ocr_tools/universal_doc_parser/core/model_factory.py

@@ -89,6 +89,19 @@ class ModelFactory:
         return recognizer
     
     @classmethod
+    def create_table_classifier(cls, config: Dict[str, Any]) -> Optional[Any]:
+        """创建表格分类器(区分有线/无线表格)"""
+        module_name = config.get('module', 'paddle')
+        
+        if module_name == 'paddle':
+            from models.adapters.paddle_table_classifier import PaddleTableClassifier
+            classifier = PaddleTableClassifier(config)
+            classifier.initialize()
+            return classifier
+        else:
+            raise ValueError(f"Unknown table classifier module: {module_name}")
+    
+    @classmethod
     def cleanup_all(cls):
         """清理所有模型资源"""
         # 在实际应用中,可以维护一个活跃模型列表进行清理