基于YOLOv5s-SE和通道剪枝的虫咬紫金蝉茶检测方法研究
戴佳兵1
宋春芳1
凌彩金2
李臻锋1
孙崇高3
1.江南大学机械工程学院/江苏省食品先进制造装备技术重点实验室,江苏无锡 2141222.广东省农业科学院茶叶研究所/广东省茶树资源创新利用重点实验室,广东广州 5106403.山东碧海包装材料有限公司,山东临沂 276600
摘要：为了实现复杂自然背景下虫咬紫金蝉茶的快速、准确识别,提出了一种基于YOLOv5s-SE和通道剪枝的虫咬紫金蝉茶检测方法.首先在YOLOv5s的主干网络中添加SE注意力机制以增强模型特征提取的能力,降低复杂背景对茶叶特征提取时的干扰;然后采用通道剪枝算法对模型进行剪枝并进行微调,实现虫咬紫金蝉茶叶片的快速、准确检测.结果表明,修剪后的模型相比原YOLOv5s模型,参数量减少60.1％,帧率提升18.6％,运算量减少29.7％,平均精度均值(mAP)为81.3％.
关键词：茶叶紫金蝉茶目标检测通道剪枝注意力机制深度学习
分类号：S126(农业物理学)S435.711(病虫害及其防治)
资助基金：以农产品为单元的广东省现代农业产业技术体系创新团队建设项目(2023KJ120)河源市科技计划项目(2021030)
论文发表日期：2024-05-15
在线出版日期：2026-09-12（本平台首次上网日期，不代表文献的发表时间）
页数：7( 157-163 )
英文信息