Function get_ocr_instance

ddddocr/api/app.py:239–281

获取或创建OCR实例

(
    config_key: str,
    ocr: bool = True,
    det: bool = False,
    old: bool = False,
    beta: bool = False,
    use_gpu: bool = False,
    device_id: int = 0,
    show_ad: bool = True,
    import_onnx_path: str = "",
    charsets_path: str = ""
)

# 函数：获取OCR实例
def get_ocr_instance(
    config_key: str,
    ocr: bool = True,
    det: bool = False,
    old: bool = False,
    beta: bool = False,
    use_gpu: bool = False,
    device_id: int = 0,
    show_ad: bool = True,
    import_onnx_path: str = "",
    charsets_path: str = ""
):
    """
    获取或创建OCR实例
    """
    if config_key in ocr_instances:
        ocr_instances[config_key]["last_used"] = time.time()
        return ocr_instances[config_key]["instance"]

    logger.info(f"创建新的OCR实例，配置: {config_key}")
    try:
        instance = DdddOcr(
            ocr=ocr,
            det=det,
            old=old,
            beta=beta,
            use_gpu=use_gpu,
            device_id=device_id,
            show_ad=show_ad,
            import_onnx_path=import_onnx_path,
            charsets_path=charsets_path
        )
        ocr_instances[config_key] = {
            "instance": instance,
            "last_used": time.time()
        }
        return instance
    except (DdddOcrInputError, InvalidImageError) as e:
        logger.error(f"创建OCR实例失败: {str(e)}")
        raise HTTPException(status_code=400, detail=str(e)) from e
    except Exception as e:
        logger.error(f"创建OCR实例失败: {str(e)}")
        raise HTTPException(status_code=500, detail=f"初始化OCR失败: {str(e)}") from e

# 清理不活跃的OCR实例
def cleanup_inactive_instances(max_idle_time: int = 3600):