病害知识库流程编排技能
核心功能
- 串联病害检测、图片嵌入、向量检索、文本检索全流程,无需人工干预
- 支持自定义编排规则,适配农作物、工业、建筑等不同场景的诊断需求
- 自动融合图文多模态检索结果,进行rerank排序,输出最匹配的诊断结果
- 支持异步批量处理,单批最大支持1000张图片的并发诊断
- 内置结果校验机制,自动过滤低置信度结果,支持人工审核回调
依赖安装
pip install pydantic>=2.0 aiohttp>=3.8 tenacity>=8.2
使用方式
单图诊断
from scripts.orchestrator import DiseasePipeline
pipeline = DiseasePipeline(config_path="./config/pipeline.yaml")
diagnosis_report = pipeline.run(
image_path="./corn_disease.jpg",
user_query="玉米叶子有黄褐色斑点,背面有霉层是什么病?"
)
批量诊断
reports = pipeline.batch_run(
image_paths=["./img1.jpg", "./img2.jpg"],
queries=["第一个图是什么病", "第二个图怎么防治"]
)
输出格式
返回完整结构化诊断报告,包含:
- 病害基本信息:名称、别名、病原、分类
- 症状描述:典型症状、易发部位
- 发生规律:发病条件、传播途径
- 防治方案:农业防治、物理防治、化学防治、生物防治
- 相似病害参考:3个易混淆病害的区分要点
- 置信度:综合诊断置信度得分(0-1)
- 处理建议:是否需要人工复核