发明公开
- 专利标题: 一种基于注意力机制的农作物病害识别方法
-
申请号: CN202410113287.2申请日: 2024-01-26
-
公开(公告)号: CN117953342A公开(公告)日: 2024-04-30
- 发明人: 岳佳欣 , 尹华国 , 陈仿雄
- 申请人: 成都农业科技职业学院
- 申请人地址: 四川省成都市温江区德通桥路392号
- 专利权人: 成都农业科技职业学院
- 当前专利权人: 成都农业科技职业学院
- 当前专利权人地址: 四川省成都市温江区德通桥路392号
- 代理机构: 贵阳中新专利商标事务所
- 代理商 石诚
- 主分类号: G06V10/82
- IPC分类号: G06V10/82 ; G06V20/10 ; G06V10/40 ; G06V10/80 ; G06N3/0464 ; G06N3/045 ; G06N3/08
摘要:
本发明公开了一种基于注意力机制的农作物病害识别方法,该方法包括有以下步骤:农作物病害数据收集及标注、数据增强、数据预处理、构建基于YOLO_V8基础架构模型、构建双分支特征融合模块BTM、构建模型整体训练损失、采用adam算法优化模型参数以及输出模型病害检测结果。用于解决现有农作物病害目标特征较小,不同病害样本不均衡,导致现有的目标检测算法存在明显的漏检和误检问题。本发明通过注意力机制强化小目标病害的特征,降低因为特征丢失导致漏检问题,同时对于其损失函数的置信度损失部分,通过引入平衡因子,使得模型在训练过程中,更加专注于对少样本的特征学习,能够有效避免由于样本不均衡导致误检的情况,从而提升算法对于病害识别的准确度。