Function object_detection_file

ddddocr/api/app.py:543–589

目标检测功能 - 接收上传的图片文件

(
    background_tasks: BackgroundTasks,
    file: UploadFile = File(...),
    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/file", response_model=DetectionResponse)
async def object_detection_file(
    background_tasks: BackgroundTasks,
    file: UploadFile = File(...),
    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="是否显示广告")
):
    """
    目标检测功能 - 接收上传的图片文件
    """
    try:
        contents = await file.read()
    except Exception as exc:
        raise HTTPException(status_code=400, detail="无法读取上传文件") from exc

    if not contents:
        raise HTTPException(status_code=400, detail="上传文件为空")
    if len(contents) > CORE_MAX_IMAGE_BYTES:
        raise HTTPException(
            status_code=400,
            detail=f"图片大小超过 {CORE_MAX_IMAGE_BYTES // 1024}KB 限制"
        )

    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=contents)
    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_match", response_model=SlideMatchResponse)