新增脚本：check_registry.py，用于检查模型注册状态及导入各个注册模块

zhch/check_registry.py

@@ -0,0 +1,55 @@
+import sys
+import os
+
+# 添加PaddleX路径
+sys.path.insert(0, '/Users/zhch158/workspace/repository.git/PaddleX')
+
+def check_registration():
+    """检查模型注册状态"""
+    
+    # 1. 检查基础注册机制
+    try:
+        from paddlex.repo_apis.base.register import MODEL_INFO_REGISTRY
+        print(f"初始注册数量: {len(MODEL_INFO_REGISTRY._table)}")
+    except Exception as e:
+        print(f"导入注册表失败: {e}")
+        return
+    
+    # 2. 手动导入各个注册模块
+    registration_modules = [
+        'paddlex.repo_apis.PaddleOCR_api.text_det.register',
+        'paddlex.repo_apis.PaddleOCR_api.text_rec.register', 
+        'paddlex.repo_apis.PaddleOCR_api.table_rec.register',
+        'paddlex.repo_apis.PaddleOCR_api.textline_orientation.register',
+        'paddlex.repo_apis.PaddleDetection_api.layout_detection.register',
+        'paddlex.repo_apis.PaddleClas_api.image_classification.register'
+    ]
+    
+    for module_name in registration_modules:
+        try:
+            __import__(module_name)
+            print(f"✓ 成功导入: {module_name}")
+            print(f"  当前注册数量: {len(MODEL_INFO_REGISTRY._table)}")
+        except Exception as e:
+            print(f"✗ 导入失败: {module_name} - {e}")
+    
+    # 3. 显示所有已注册模型
+    print(f"\n最终已注册的模型({len(MODEL_INFO_REGISTRY._table)}):")
+    for model_name in sorted(MODEL_INFO_REGISTRY._table.keys()):
+        print(f"  - {model_name}")
+    
+    # 4. 检查SLANet_plus是否存在
+    if 'SLANet_plus' in MODEL_INFO_REGISTRY._table:
+        print(f"\n✓ SLANet_plus 已注册!")
+        model_info = MODEL_INFO_REGISTRY._table['SLANet_plus']
+        print(f"  配置路径: {model_info.get('config_path', 'N/A')}")
+    else:
+        print(f"\n✗ SLANet_plus 未注册")
+        # 查找相似模型
+        similar_models = [name for name in MODEL_INFO_REGISTRY._table.keys() 
+                         if 'sla' in name.lower() or 'table' in name.lower()]
+        if similar_models:
+            print(f"  相似模型: {similar_models}")
+
+if __name__ == "__main__":
+    check_registration()