Function slide_match_recognition

ddddocr/api/app.py:593–633

滑块验证码匹配 - 接收Base64编码的目标图和背景图

(
    request: SlideMatchRequest,
    background_tasks: BackgroundTasks,
    ocr: bool = Query(False, description="是否启用OCR功能"),
    det: bool = Query(False, description="是否启用目标检测功能"),
    use_gpu: bool = Query(default_use_gpu, description="是否使用GPU加速"),
    device_id: int = Query(default_device_id, description="GPU设备ID"),
    show_ad: bool = Query(default_show_ad, description="是否显示广告")
)

# 滑块匹配端点
@app.post("/slide_match", response_model=SlideMatchResponse)
async def slide_match_recognition(
    request: SlideMatchRequest,
    background_tasks: BackgroundTasks,
    ocr: bool = Query(False, description="是否启用OCR功能"),
    det: bool = Query(False, description="是否启用目标检测功能"),
    use_gpu: bool = Query(default_use_gpu, description="是否使用GPU加速"),
    device_id: int = Query(default_device_id, description="GPU设备ID"),
    show_ad: bool = Query(default_show_ad, description="是否显示广告")
):
    """
    滑块验证码匹配 - 接收Base64编码的目标图和背景图
    """
    target_data = _decode_base64_bytes(request.target_image)
    background_data = _decode_base64_bytes(request.background_image)

    config_key = f"ocr={ocr}-det={det}-gpu={use_gpu}-dev={device_id}"

    ocr_instance = get_ocr_instance(
        config_key, ocr, det, False, False, use_gpu, device_id, show_ad,
        default_import_onnx_path, default_charsets_path
    )

    start_time = time.time()
    try:
        result = ocr_instance.slide_match(
            target_bytes=target_data,
            background_bytes=background_data,
            simple_target=request.simple_target,
            flag=request.flag
        )
    except (DdddOcrInputError, InvalidImageError) as exc:
        raise HTTPException(status_code=400, detail=str(exc)) from exc
    except Exception as exc:
        logger.error(f"滑块匹配失败: {str(exc)}")
        raise HTTPException(status_code=500, detail=f"滑块匹配失败: {str(exc)}") from exc

    background_tasks.add_task(cleanup_inactive_instances)
    return {
        "result": result,
        "processing_time": time.time() - start_time
    }

# 滑块比较端点
@app.post("/slide_comparison", response_model=SlideComparisonResponse)