PaddleX/zhch/check_registry.py

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()