| 123456789101112131415161718192021222324 |
- """
- 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 加速
- }
|
