"""
OCR识别配置
"""

# OCR配置 - 使用llama-server API
OCR_CONFIG = {
    'use_llama_api': True,          # 使用llama-server API
    'api_host': 'localhost',        # API服务器地址
    'api_port': 8111,               # API端口
    'model': 'PaddleOCR-VL-1.5-GGUF.gguf',  # 模型名称
    'prompt': '请识别图片中的数字编号，只返回数字，不要其他内容',  # OCR提示词
    'temperature': 0.3,             # 温度参数
    'timeout': 30,                  # 超时时间(秒)
}

# 人体分割模型配置
SEGMENTATION_CONFIG = {
    # 模型路径 - 支持 RKNN 格式 (RK3588 平台)
    'model_path': '/home/admin/dsh/testrk3588/yolov8n-seg.rknn',
    'model_type': 'rknn',           # 模型类型: 'rknn'
    'input_size': (640, 640),       # 模型输入尺寸
    'conf_threshold': 0.5,          # 分割置信度阈值
    'use_npu': True,                # 使用 NPU 加速
}