基于改进YOLOX-Nano的农作物叶片病害检测与识别方法
作者机构:华南农业大学数学与信息学院广东广州510642 东莞城市学院人工智能学院广东东莞523419 深圳信息职业技术学院广东深圳518172
基 金:国家自然科学基金(61573157) 广东省基础与应用基础研究基金(2022A1515011447) 广东省教育厅重点领域专项(2021ZDZX1029)
出 版 物:《华南农业大学学报》 (Journal of South China Agricultural University)
年 卷 期:2023年第44卷第4期
页 码:593-603页
摘 要:【目的】实现精确迅速的农作物病害检测,减少人工诊断成本,降低病害带来的农作物产量和品质影响。【方法】根据对农作物病害和病斑特征的分析,提出一种基于卷积注意力机制改进的YOLOX-Nano智能检测与识别模型,该模型采用CSPDarkNet作为主干网络,将卷积注意力模块CBAM引入到YOLOX-Nano网络结构的特征金字塔(Feature pyramid network,FPN)中,并在训练中引入Mixup数据增强方式,同时将分类的损失函数由二分类交叉熵损失函数(Binary cross entropy loss,BCE Loss)替换为焦点损失函数Focal Loss、回归损失函数由GIOU Loss替换为本文设计的CenterIOU Loss函数,采用迁移学习策略训练改进的YOLOX-Nano模型,以此提升农作物病害检测的精度。【结果】改进后的YOLOX-Nano模型仅有0.98×10^(6)的参数量,在移动端测试单张图片检测时间约为0.187 s,平均识别精度达到99.56%。实践结果表明,其能快速有效地检测与识别苹果、玉米、葡萄、草莓、马铃薯和番茄等农作物的常见病害,且达到了精度与速度的平衡。【结论】改进后的模型不仅对农作物叶片病害识别具有较高的精度和较快的检测速度,参数量和计算量较少,还易于部署在手机等移动端设备。该模型实现了在田间复杂环境对多种农作物病害精准定位与识别,对于指导早期农作物病害的防治具有十分重要的现实意义。
主 题 词:YOLOX-Nano网络 病害识别 Focal Loss 注意力机制 农作物病害
学科分类:08[工学] 080203[080203] 09[农学] 0904[农学-动物医学类] 090401[090401] 0802[工学-机械学] 090402[090402]
核心收录:
D O I:10.7671/j.issn.1001-411X.202207039
馆 藏 号:203122392...