|
|
@@ -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):
|
|
|
"""清理所有模型资源"""
|
|
|
# 在实际应用中,可以维护一个活跃模型列表进行清理
|