PaddleX/zhch/test_evaluator.py

# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#    http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.


import sys
import os
from paddlex.engine import Engine

# 设置环境变量
# os.environ["PADDLE_PDX_PADDLEOCR_PATH"] = "/Users/zhch158/workspace/repository.git/PaddleX/paddle_env/lib/python3.11/site-packages/paddleocr"

# 添加PaddleX路径
# sys.path.insert(0, '/Users/zhch158/workspace/repository.git/PaddleX')

def force_import_registrations():
    """强制导入所有注册模块"""
    try:
        # 强制导入注册模块
        import paddlex.repo_apis.PaddleOCR_api.table_rec.register
        import paddlex.repo_apis.PaddleOCR_api.text_rec.register  
        import paddlex.repo_apis.PaddleOCR_api.text_det.register
        import paddlex.repo_apis.PaddleOCR_api.formula_rec.register
        # import paddlex.repo_apis.PaddleOCR_api.textline_orientation.register
        # import paddlex.repo_apis.PaddleOCR_api.doc_text_orientation.register
        
        print("✓ 所有注册模块导入成功")
        
        # 检查注册状态
        from paddlex.repo_apis.base.register import MODEL_INFO_REGISTRY
        table_models = [name for name in MODEL_INFO_REGISTRY._table.keys() 
                       if 'SLA' in name]
        print(f"已注册的表格模型: {table_models}")
        
        if 'SLANet_plus' in MODEL_INFO_REGISTRY._table:
            print("✓ SLANet_plus 已成功注册")
            return True
        else:
            print("✗ SLANet_plus 仍未注册")
            return False
            
    except Exception as e:
        print(f"注册模块导入失败: {e}")
        return False

if __name__ == "__main__":
    # if force_import_registrations():
    #     # 现在运行引擎
    #     from paddlex.engine import Engine
    #     print("开始运行...")
    #     Engine().run()
    # else:
    #     print("注册失败，无法继续")
    Engine().run()