Function object_detection

ddddocr/api/app.py:501–539

目标检测功能 - 接收Base64编码的图片

(
    request: Base64Image,
    background_tasks: BackgroundTasks,
    ocr: bool = Query(False, description="是否启用OCR功能"),
    det: bool = Query(True, 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("/det", response_model=DetectionResponse)
async def object_detection(
    request: Base64Image,
    background_tasks: BackgroundTasks,
    ocr: bool = Query(False, description="是否启用OCR功能"),
    det: bool = Query(True, 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编码的图片
    """
    try:
        img_data = _decode_base64_bytes(request.image)
    except HTTPException:
        raise

    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.detection(img_bytes=img_data)
    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("/det/file", response_model=DetectionResponse)