首页 | 本学科首页   官方微博 | 高级检索  
     检索      

基于改进YOLOv5s的番茄叶片病害检测方法
引用本文:陶兆胜,石鑫宇,王勇,伍毅,吴浩.基于改进YOLOv5s的番茄叶片病害检测方法[J].沈阳农业大学学报,2023(6):712-721.
作者姓名:陶兆胜  石鑫宇  王勇  伍毅  吴浩
作者单位:安徽工业大学机械工程学院
基金项目:安徽省自然科学基金面上项目(2108085ME166);;安徽高校自然科学研究项目重点项目(KJ2021A0408);
摘    要:为了解决现有的农作物病害检测方法对不同番茄叶片病害检测的精度低、效果差的问题,提出一种基于YOLOv5网络模型改进的番茄叶片病害检测模型YOLOv5s-TLD。首先在原YOLOv5s模型的Backbone中构建DCAM注意力机制模块,通过制定双通道注意力和空间注意力机制加强模型对番茄叶片病理特征的提取能力,并减弱模型受复杂背景特征的影响,以提高模型对不同种类病害的检测精度和分类精度;然后应用融合Swin Transformer的C3STR模块替换原网络第6层的C3模块,强化模型在多尺度上建模的能力,实现模型对小尺寸的番茄叶片病害残差特征的高精度学习;再运用BiFPN加权双向特征金字塔网络替换原YOLOv5模型Head的PANet路径聚合网络,该网络采用跨尺度特征融合和可学习权重的方式融合模型不同层次的特征,在增强网络的特征融合能力的同时使网络获得更多的特征信息,以提高模型的感受野和特征表达能力;最后进行不同模型的检测对比试验,并在实际复杂场景下进行番茄叶片病害检测试验。试验结果表明:YOLOv5s-TLD模型平均精度均值和召回率分别为97.7%和96.3%,较原YOLOv5s模型平均精...

关 键 词:深度学习  YOLOv5s  卷积神经网络  病害检测  番茄叶片病害
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号